From b899d80c8a641ef636a99e55a52bf508292c5e84 Mon Sep 17 00:00:00 2001 From: StrangerZhang Date: Fri, 7 Jun 2019 20:34:33 +0800 Subject: [PATCH] add local_rank to train.py --- tools/train.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tools/train.py b/tools/train.py index 46408736..a217ac09 100644 --- 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()