From b0158a593e1692e535f18ae79db490f3b58c3210 Mon Sep 17 00:00:00 2001
From: Ross Wightman <rwightman@gmail.com>
Date: Fri, 5 Apr 2019 20:49:58 -0700
Subject: [PATCH] Fix distributed train script

---
 distributed_train.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/distributed_train.sh b/distributed_train.sh
index a1c68c0a..88406997 100755
--- a/distributed_train.sh
+++ b/distributed_train.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
 NUM_PROC=$1
 shift
-python -m torch.distributed.launch --nproc_per_node=$NUM_PROC dtrain.py "$@"
+python -m torch.distributed.launch --nproc_per_node=$NUM_PROC train.py "$@"