stablediffusion吧 关注:28,043贴子:97,321
  • 0回复贴,共1

炼丹时,优化器选择lion就出错

只看楼主收藏回复

我用的秋叶大佬的一键包,但在炼丹的时候,优化器,选择“”没有问题
但如果选择lion就会有问题,应该怎么解决?
配置如下:
pretrained_model_name_or_path = "D:/SDA/models/Stable-diffusion/chilloutmix_NiPrunedFp32Fix.safetensors"
train_data_dir = "D:/lora-scripts-v1.5.1/train/doudou"
resolution = "512,512"
enable_bucket = true
min_bucket_reso = 256
max_bucket_reso = 1_024
output_name = "doudou"
output_dir = "./output"
save_model_as = "safetensors"
save_every_n_epochs = 2
max_train_epochs = 10
train_batch_size = 1
network_train_unet_only = false
network_train_text_encoder_only = false
learning_rate = 0.0001
unet_lr = 0.0001
text_encoder_lr = 0.00001
lr_scheduler = "cosine_with_restarts"
optimizer_type = "Lion"
lr_scheduler_num_cycles = 1
network_module = "networks.lora"
network_dim = 32
network_alpha = 32
logging_dir = "./logs"
caption_extension = ".txt"
shuffle_caption = true
keep_tokens = 0
max_token_length = 255
seed = 1_337
prior_loss_weight = 1
clip_skip = 2
mixed_precision = "fp16"
save_precision = "fp16"
xformers = true
cache_latents = true
persistent_data_loader_workers = true
lr_warmup_steps = 0


IP属地:福建1楼2023-09-13 11:07回复