from local

This commit is contained in:
dohe0342 2023-03-23 23:03:00 +09:00
parent 529f9faba2
commit 8436c2c44b
2 changed files with 1 additions and 1 deletions

View File

@ -1817,7 +1817,7 @@ def run_adapter_uda(rank, world_size, args, wb=None):
optimizer=optimizer, optimizer=optimizer,
scheduler=scheduler, scheduler=scheduler,
sp=sp, sp=sp,
train_dl=train_dl, train_dl=[train_dl, train_dl_uda],
valid_dl=valid_dl, valid_dl=valid_dl,
scaler=scaler, scaler=scaler,
tb_writer=tb_writer, tb_writer=tb_writer,