diff options
Diffstat (limited to 'Codes/train.sh')
| -rw-r--r-- | Codes/train.sh | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/Codes/train.sh b/Codes/train.sh new file mode 100644 index 0000000..3145aa9 --- /dev/null +++ b/Codes/train.sh @@ -0,0 +1,6 @@ +python train.py --dataset kaulsdorf_gray \ + --train_folder /home/lens/Desktop/thumbs/kaulsdorf_gray/Train/ \ + --test_folder /home/lens/Desktop/thumbs/kaulsdorf_gray/Test/ \ + --gpu 0 \ + --iters 500000 + |