commit | e5f1ae536432b08e4880070b69261b14ba831c0d | [log] [tgz] |
---|---|---|
author | nailixing <xingnaili14@gmail.com> | Sun May 16 11:39:56 2021 +0800 |
committer | nailixing <xingnaili14@gmail.com> | Sun May 16 11:39:56 2021 +0800 |
tree | 01e33f029b886865cf097bf8cdbd561354f66026 | |
parent | 4d7eccca3abe652583d96b156c303568bbdde90c [diff] |
script for distributed training of cifar-10
diff --git a/examples/cifar_distributed_cnn/run-rtx.sh b/examples/cifar_distributed_cnn/run-rtx.sh new file mode 100644 index 0000000..113711c --- /dev/null +++ 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 +