run on GPU ermongroup#1

AmmarkoV · Oct 13, 2022 · 0f64899 · 0f64899
1 parent ce95a3f
commit 0f64899
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.sh b/train.sh
@@ -14,7 +14,7 @@ else
  EXPERIMENT=$2
 fi
 
-PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python python main.py --config $CONFIGURATION --doc $EXPERIMENT
+CUDA_VISIBLE_DEVICES="1" PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python python main.py --config $CONFIGURATION --doc $EXPERIMENT
 
 cd "exp/logs"
 scp -P 2222 -r $EXPERIMENT/ [email protected]:/home/ammar/public_html/ncsnv2_archimedes/exp/logs/