from local

This commit is contained in:
dohe0342 2023-02-03 01:24:19 +09:00
parent 93e4e63050
commit 846af93220
2 changed files with 3 additions and 0 deletions

View File

@ -715,6 +715,9 @@ def main():
else: args.exp_dir = Path(args.exp_dir)
args.lang_dir = Path(args.lang_dir)
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: