From e5f1ae536432b08e4880070b69261b14ba831c0d Mon Sep 17 00:00:00 2001 From: nailixing Date: Sun, 16 May 2021 11:39:56 +0800 Subject: [PATCH] script for distributed training of cifar-10 --- examples/cifar_distributed_cnn/run-rtx.sh | 3 +++ 1 file changed, 3 insertions(+) create mode 100644 examples/cifar_distributed_cnn/run-rtx.sh diff --git a/examples/cifar_distributed_cnn/run-rtx.sh b/examples/cifar_distributed_cnn/run-rtx.sh new file mode 100644 index 000000000..113711c43 --- /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 +