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

Fix args problem

parent b286dd0d
No related branches found
No related tags found
No related merge requests found
......@@ -81,7 +81,7 @@ def main():
trainer = pl.Trainer(accelerator="gpu",
devices=num_gpus,
profiler="simple" if args.proffiler else None,
profiler="simple" if args.profiler else None,
default_root_dir="./logs",
logger=WandbLogger(project="slot-att", offline=True) if args.wandb else None,
strategy="ddp_find_unused_parameters_true" if num_gpus > 1 else "auto",
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment