from local

This commit is contained in:
dohe0342 2023-01-13 16:51:44 +09:00
parent 04977d959a
commit da288af854
2 changed files with 1 additions and 1 deletions

View File

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