summaryrefslogtreecommitdiff
path: root/scripts/train_512p_multigpu.sh
diff options
context:
space:
mode:
Diffstat (limited to 'scripts/train_512p_multigpu.sh')
-rwxr-xr-xscripts/train_512p_multigpu.sh2
1 files changed, 2 insertions, 0 deletions
diff --git a/scripts/train_512p_multigpu.sh b/scripts/train_512p_multigpu.sh
new file mode 100755
index 0000000..16f0a1a
--- /dev/null
+++ b/scripts/train_512p_multigpu.sh
@@ -0,0 +1,2 @@
+######## Multi-GPU training example #######
+python train.py --name label2city_512p --batchSize 8 --gpu_ids 0,1,2,3,4,5,6,7 \ No newline at end of file