diff --git a/train.ps1 b/train.ps1 index bf50984..45b8b2a 100644 --- a/train.ps1 +++ b/train.ps1 @@ -52,7 +52,7 @@ $multi_gpu = 0 # multi gpu | 多显卡训练 该参数仅限在显卡数 >= 2 $lowram = 0 # lowram mode | 低内存模式 该模式下会将 U-net 文本编码器 VAE 转移到 GPU 显存中 启用该模式可能会对显存有一定影响 # 优化器设置 -$optimizer_type = "AdamW8bit" # Optimizer type | 优化器类型 默认为 8bitadam,可选:AdamW AdamW8bit Lion SGDNesterov SGDNesterov8bit DAdaptation AdaFactor +$optimizer_type = "AdamW8bit" # Optimizer type | 优化器类型 默认为 AdamW8bit,可选:AdamW AdamW8bit Lion SGDNesterov SGDNesterov8bit DAdaptation AdaFactor # LyCORIS 训练设置 $algo = "lora" # LyCORIS network algo | LyCORIS 网络算法 可选 lora、loha、lokr、ia3、dylora。lora即为locon diff --git a/train.sh b/train.sh index 56b4421..5383259 100644 --- a/train.sh +++ b/train.sh @@ -51,7 +51,7 @@ persistent_data_loader_workers=0 # persistent dataloader workers | 容易爆内 clip_skip=2 # clip skip | 玄学 一般用 2 # 优化器设置 -optimizer_type="AdamW8bit" # Optimizer type | 优化器类型 默认为 8bitadam,可选:AdamW AdamW8bit Lion SGDNesterov SGDNesterov8bit DAdaptation AdaFactor +optimizer_type="AdamW8bit" # Optimizer type | 优化器类型 默认为 AdamW8bit,可选:AdamW AdamW8bit Lion SGDNesterov SGDNesterov8bit DAdaptation AdaFactor # LyCORIS 训练设置 algo="lora" # LyCORIS network algo | LyCORIS 网络算法 可选 lora、loha、lokr、ia3、dylora。lora即为locon