Skip to content
Snippets Groups Projects
Commit 1a36bb26 authored by Alexandre Chapin's avatar Alexandre Chapin :race_car:
Browse files

Set wandb log

parent 349fae70
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 1 addition and 1 deletion
train_sa.py
+ 1
− 1
View file @ 1a36bb26
......@@ -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,
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment