1 files changed, 5 insertions, 3 deletions
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