from local

This commit is contained in:
dohe0342 2022-12-10 13:40:08 +09:00
parent b98df53225
commit 89b2192954
2 changed files with 1 additions and 1 deletions

View File

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