From da65481b66e3e2ae93c48d537c0571b5f9dc95ef Mon Sep 17 00:00:00 2001
From: darkliang <12132342@mail.sustech.edu.cn>
Date: Tue, 19 Mar 2024 15:20:41 +0800
Subject: [PATCH] compatible with torch 2.0

---
 opengait/main.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/opengait/main.py b/opengait/main.py
index 8358d6d..087edcb 100644
--- a/opengait/main.py
+++ b/opengait/main.py
@@ -9,6 +9,8 @@ from utils import config_loader, get_ddp_module, init_seeds, params_count, get_m
 parser = argparse.ArgumentParser(description='Main program for opengait.')
 parser.add_argument('--local_rank', type=int, default=0,
                     help="passed by torch.distributed.launch module")
+parser.add_argument('--local-rank', type=int, default=0,
+                    help="passed by torch.distributed.launch module, for pytorch >=2.0")
 parser.add_argument('--cfgs', type=str,
                     default='config/default.yaml', help="path of config file")
 parser.add_argument('--phase', default='train',