from local

This commit is contained in:
dohe0342 2023-01-13 16:47:15 +09:00
parent 2241ba64e2
commit 4d1ace8306
2 changed files with 3 additions and 0 deletions

View File

@ -1192,6 +1192,9 @@ def main():
logging.info("save arguments to config.yaml...")
save_args(args)
if args.wandb: wb = wandb.init(project="d2v-T", entity="dohe0342", config=vars(args))
else: wb = None
world_size = args.world_size
assert world_size >= 1
if world_size > 1: