train.sh


1
2
3
4
5
6
7
8
9
10
11

#!/bin/bash

if [ "$1" == "" ]; then
    echo "Usage: $0 [dataset]"
    exit 1
fi
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