From 1e34fe2bbee136a325501f43fd80f19212d952d1 Mon Sep 17 00:00:00 2001 From: alexcbb <alexchapin@hotmail.fr> Date: Fri, 28 Jul 2023 13:34:57 +0200 Subject: [PATCH] Find unused params --- train_sa.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_sa.py b/train_sa.py index 5963cf5..0397a7c 100644 --- a/train_sa.py +++ b/train_sa.py @@ -84,7 +84,7 @@ def main(): profiler="simple", default_root_dir="./logs", logger=WandbLogger(project="slot-att", offline=True) if args.wandb else None, - strategy="ddp" if num_gpus > 1 else "auto", + strategy="ddp_find_unused_parameters_true" if num_gpus > 1 else "auto", callbacks=[checkpoint_callback, early_stopping], log_every_n_steps=100, val_check_interval=cfg["training"]["validate_every"], -- GitLab