add local_rank to train.py

STVIR · Jun 7, 2019 · b899d80 · b899d80
1 parent 68c4667
commit b899d80
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tools/train.py b/tools/train.py
@@ -39,6 +39,8 @@
                     help='configuration of tracking')
 parser.add_argument('--seed', type=int, default=123456,
                     help='random seed')
+parser.add_argument('--local_rank', type=int, default=0,
+                    help='compulsory for pytorch launcer')
 args = parser.parse_args()