from local

This commit is contained in:
dohe0342 2022-12-10 14:29:54 +09:00
parent 46090c08b1
commit e440c9f31a
3 changed files with 1 additions and 0 deletions

View File

@ -1185,6 +1185,7 @@ def run(rank, world_size, args, wb=None):
logging.info(f"enc name len = {len(enc_names)}")
logging.info(f"enc param len = {len(enc_param)}")
logging.info(type(model.parameters()))
optimizer_enc = ScaledAdam(
enc_param,