from local

This commit is contained in:
dohe0342 2023-02-03 01:25:36 +09:00
parent 03a917f7ce
commit 223558f0c1
2 changed files with 1 additions and 1 deletions

View File

@ -582,7 +582,7 @@ def train_one_epoch(
params.best_train_loss = params.train_loss
def run(rank, world_size, args):
def run(rank, world_size, args, wb=None):
"""
Args:
rank: