Mirrors
/
EveryDream2trainer
mirror of https://github.com/victorchall/EveryDream2trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
parent eea899f4a0 ba5706c5d8
commit 4433bd7806
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -588,7 +588,7 @@ 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",