fantacaty fantacaty
关注数: 22 粉丝数: 23 发帖数: 1,717 关注贴吧数: 20
求大神帮忙解答,为啥用秋叶的一键lora包,为啥慢的离谱? 显卡是tesla M409,具体参数如下: model_train_type = "sd-lora"pretrained_model_name_or_path = "./sd-models/v1-5-pruned.ckpt"v2 = falsetrain_data_dir = "./train/sheep7682"prior_loss_weight = 1resolution = "768,768"enable_bucket = falsemin_bucket_reso = 256max_bucket_reso = 1_024bucket_reso_steps = 64output_name = "sheep7682"output_dir = "./output"save_model_as = "safetensors"save_precision = "fp16"save_every_n_epochs = 2max_train_epochs = 20train_batch_size = 6gradient_checkpointing = falsenetwork_train_unet_only = falsenetwork_train_text_encoder_only = falselearning_rate = 0.0001unet_lr = 0.0001text_encoder_lr = 0.00005lr_scheduler = "cosine_with_restarts"lr_warmup_steps = 0lr_scheduler_num_cycles = 1optimizer_type = "Lion"network_module = "networks.lora"network_dim = 64network_alpha = 32network_dropout = 0log_with = "tensorboard"logging_dir = "./logs"caption_extension = ".txt"shuffle_caption = trueweighted_captions = falsekeep_tokens = 0max_token_length = 255seed = 1_337clip_skip = 2mixed_precision = "fp16"full_fp16 = falsefull_bf16 = falsexformers = truelowram = falsecache_latents = truecache_latents_to_disk = truecache_text_encoder_outputs = falsecache_text_encoder_outputs_to_disk = falsepersistent_data_loader_workers = truemulti_gpu = false Lion模型,768*768图片,30张,5 repeat,20epochs,bs为6. 显卡需要换成tcc吗,问题是之前用秋叶的老模型 速度还可以啊!
1 下一页