From e304c5cc08e4b168090b3a8ac24b7a83315f0407 Mon Sep 17 00:00:00 2001
From: aliencyc <40140686+aliencyc@users.noreply.github.com>
Date: Sun, 13 Jun 2021 20:21:31 +0800
Subject: [PATCH] Update 6_train_SAEHD.sh

---
 scripts/6_train_SAEHD.sh | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/scripts/6_train_SAEHD.sh b/scripts/6_train_SAEHD.sh
index 10ea707ef..fae493fca 100755
--- a/scripts/6_train_SAEHD.sh
+++ b/scripts/6_train_SAEHD.sh
@@ -6,5 +6,6 @@ $DFL_PYTHON "$DFL_SRC/main.py" train \
     --training-data-dst-dir "$DFL_WORKSPACE/data_dst/aligned" \
     --pretraining-data-dir "$DFL_ROOT/pretrain_CelebA" \
     --model-dir "$DFL_WORKSPACE/model" \
-    --model SAEHD
-
+    --model SAEHD \
+    --force-model-name DFD224F \
+    --force-gpu-idxs 0