From 1a36bb264a37c25eb1f7b3d57664b7f8f9e06182 Mon Sep 17 00:00:00 2001
From: alexcbb <alexchapin@hotmail.fr>
Date: Mon, 24 Jul 2023 15:39:16 +0200
Subject: [PATCH] Set wandb log
---
train_sa.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/train_sa.py b/train_sa.py
index 5b11bed..17886ac 100644
--- a/train_sa.py
+++ b/train_sa.py
@@ -60,7 +60,7 @@ def main():
#### Create model
model = LitSlotAttentionAutoEncoder(resolution, num_slots, num_iterations, cfg=cfg)
- wandb_logger = WandbLogger()
+ wandb_logger = WandbLogger(project="slot-attention")
checkpoint_callback = ModelCheckpoint(
save_top_k=10,
--
GitLab