summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--.gitignore2
-rwxr-xr-xrun-flow.sh7
2 files changed, 5 insertions, 4 deletions
diff --git a/.gitignore b/.gitignore
index 7290625..ef552b4 100644
--- a/.gitignore
+++ b/.gitignore
@@ -42,3 +42,5 @@ test/data/legacy_serialized.pt
*~
.idea
recursive/
+*.swp
+
diff --git a/run-flow.sh b/run-flow.sh
index 858be31..3f650dc 100755
--- a/run-flow.sh
+++ b/run-flow.sh
@@ -6,7 +6,7 @@ dataset="woodflowfixed"
# --fold_A "$HOME/Desktop/thumbs/$dataset/A" \
# --fold_B "$HOME/Desktop/thumbs/$dataset/B" \
# --fold_AB "./datasets/$dataset/"
-#
+
python train.py \
--dataroot "./datasets/$dataset" \
--name "$dataset" \
@@ -17,11 +17,10 @@ python train.py \
--which_direction AtoB \
--lambda_B 100 \
--dataset_mode aligned \
+ --continue_train \
+ --which_epoch latest \
--no_lsgan --norm batch --pool_size 0
-# --continue_train \
-# --which_epoch latest \
-
#python canny-cv.py \
# --dataroot /home/ubuntu/Desktop/thumbs/wood/B/train/ \
# --name woodcanny \