From d301aa098165cc54af3552d8534054cf18551509 Mon Sep 17 00:00:00 2001 From: shayaharon Date: Tue, 7 May 2024 15:03:10 +0300 Subject: [PATCH] more deprecated default args removed from LaunchConfig --- .../training/utils/distributed_training_utils.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/src/super_gradients/training/utils/distributed_training_utils.py b/src/super_gradients/training/utils/distributed_training_utils.py index c4dbd55e80..6d587466a1 100755 --- a/src/super_gradients/training/utils/distributed_training_utils.py +++ b/src/super_gradients/training/utils/distributed_training_utils.py @@ -10,7 +10,6 @@ from torch import distributed as dist from torch.cuda.amp import autocast from torch.distributed import get_rank, all_gather_object -from torch.distributed.elastic.multiprocessing import Std from torch.distributed.elastic.multiprocessing.errors import record from torch.distributed.launcher.api import LaunchConfig, elastic_launch @@ -345,9 +344,6 @@ def restart_script_with_ddp(num_gpus: int = None): max_restarts=0, monitor_interval=5, start_method="spawn", - log_dir=None, - redirects=Std.NONE, - tee=Std.NONE, metrics_cfg={}, )