From 58f998e10d00e455affcaf0af21fa8aeb67f328b Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Tue, 12 Jun 2018 17:29:09 +0200
Subject: zeroes

---
 train.sh | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

(limited to 'train.sh')

diff --git a/train.sh b/train.sh
index a3f7460..df4969e 100755
--- a/train.sh
+++ b/train.sh
@@ -2,6 +2,8 @@ if [ "$1" == "" ]; then
     echo "Usage: $0 [dataset]"
     exit 1
 fi
-
-python train.py --dataroot "./datasets/${1}/" --name "$1" --label_nc 0 --no_instance --continue_train --which_epoch latest
-
+if [ -e "./checkpoints/${1}" ]; then
+  python train.py --dataroot "./datasets/${1}/" --name "$1" --label_nc 0 --no_instance --continue_train --which_epoch latest
+else
+  python train.py --dataroot "./datasets/${1}/" --name "$1" --label_nc 0 --no_instance
+fi
-- 
cgit v1.2.3-70-g09d2