From a068578e1e8a04e9020cb8811d200b64dbfce15a Mon Sep 17 00:00:00 2001
From: Akiba <akiba@anzu.link>
Date: Sat, 22 Apr 2023 17:36:34 +0800
Subject: [PATCH] fix DAdaptation

---
 train.ps1 | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/train.ps1 b/train.ps1
index 45b8b2a..9cef6ea 100644
--- 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")