from local
This commit is contained in:
parent
e65559b166
commit
0c43826323
Binary file not shown.
@ -1183,10 +1183,6 @@ def run(rank, world_size, args, wb=None):
|
||||
dec_names.append(n)
|
||||
dec_param.append(p)
|
||||
|
||||
logging.info(f"enc name len = {len(enc_names)}")
|
||||
logging.info(f"enc param len = {len(enc_param)}")
|
||||
#logging.info(model.parameters())
|
||||
|
||||
optimizer_enc = ScaledAdam(
|
||||
*enc_param,
|
||||
lr=params.peak_enc_lr,
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user