from local
This commit is contained in:
parent
85697a7370
commit
fef51784e4
Binary file not shown.
@ -1055,6 +1055,8 @@ def train_one_epoch(
|
|||||||
is_training=True,
|
is_training=True,
|
||||||
decode = True if batch_idx % params.decode_interval == 0 else False,
|
decode = True if batch_idx % params.decode_interval == 0 else False,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
print(loss_info)
|
||||||
try: loss_info.reduce(loss.device)
|
try: loss_info.reduce(loss.device)
|
||||||
except: pass
|
except: pass
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user