fix DAdaptation

2023-04-22 17:36:34 +08:00 · 2023-04-22 17:36:34 +08:00 · a068578e1e
parent e86151792d
commit a068578e1e
1 changed files with 4 additions and 0 deletions
--- a/train.ps1
+++ b/train.ps1
@ -107,6 +107,10 @@ if ($optimizer_type) {
  [void]$ext_args.Add("--optimizer_type=" + $optimizer_type)
 }

+if ($optimizer_type -eq "DAdaptation") {
+  [void]$ext_args.Add("--optimizer_args decouple=True")
+}
+
 if ($network_module -eq "lycoris.kohya") {
  [void]$ext_args.Add("--network_args")
  [void]$ext_args.Add("conv_dim=$conv_dim")