from local

This commit is contained in:
dohe0342 2022-12-10 13:39:17 +09:00
parent 91b9cc29a7
commit 8afcd618c6
2 changed files with 3 additions and 0 deletions

View File

@ -1344,6 +1344,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: