Merge pull request #19 from janekm/wandb_fix

Fix wandb setup
This commit is contained in:
Victor Hall 2023-01-23 17:38:49 -08:00 committed by GitHub
commit 4433bd7806
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -588,8 +588,8 @@ def main(args):
if args.wandb is not None and args.wandb: if args.wandb is not None and args.wandb:
wandb.init(project=args.project_name, sync_tensorboard=True, ) wandb.init(project=args.project_name, sync_tensorboard=True, )
else:
log_writer = SummaryWriter(log_dir=log_folder, log_writer = SummaryWriter(log_dir=log_folder,
flush_secs=5, flush_secs=5,
comment="EveryDream2FineTunes", comment="EveryDream2FineTunes",
) )