from local
This commit is contained in:
parent
f0a939469e
commit
573a79b08d
Binary file not shown.
@ -1229,8 +1229,8 @@ def run(rank, world_size, args, wb=None):
|
|||||||
dec_names.append(n)
|
dec_names.append(n)
|
||||||
dec_param.append(p)
|
dec_param.append(p)
|
||||||
|
|
||||||
logging.info(enc_names)
|
if rank == 0:
|
||||||
|
print(enc_names)
|
||||||
optimizer_enc = ScaledAdam(
|
optimizer_enc = ScaledAdam(
|
||||||
enc_param,
|
enc_param,
|
||||||
lr=params.peak_enc_lr,
|
lr=params.peak_enc_lr,
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user