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