seed (#370)

Author: lyuwenyu

rtdetr_pytorch/src/misc/dist.py

@@ -182,7 +182,6 @@ def sync_time():
 
 def set_seed(seed):
     # fix the seed for reproducibility
-    seed = seed + get_rank()
     torch.manual_seed(seed)
     np.random.seed(seed)
     random.seed(seed)

rtdetr_pytorch/tools/train.py

@@ -15,6 +15,8 @@ def main(args, ) -> None:
     '''main
     '''
     dist.init_distributed()
+    if args.seed is not None:
+        dist.set_seed(args.seed)
 
     assert not all([args.tuning, args.resume]), \
         'Only support from_scrach or resume or tuning at one time'
@@ -42,7 +44,7 @@ def main(args, ) -> None:
     parser.add_argument('--tuning', '-t', type=str, )
     parser.add_argument('--test-only', action='store_true', default=False,)
     parser.add_argument('--amp', action='store_true', default=False,)
-
+    parser.add_argument('--seed', type=int, help='seed',)
     args = parser.parse_args()
 
     main(args)