script for distributed training of cifar-10

GLKong · May 16, 2021 · e5f1ae5 · e5f1ae5
1 parent 4d7eccc
commit e5f1ae5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/examples/cifar_distributed_cnn/run-rtx.sh b/examples/cifar_distributed_cnn/run-rtx.sh
@@ -0,0 +1,3 @@
+#!/usr/bin/env python -W ignore::DeprecationWarning
+mpiexec -np 2 python train_cnn.py resnet cifar10 -l 0.015 -b 32
+