mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
ea1e0c0f67
commit
fef78e281f
Binary file not shown.
@ -1115,11 +1115,11 @@ def train_one_epoch(
|
|||||||
f"grad_scale is too small, exiting: {cur_grad_scale}"
|
f"grad_scale is too small, exiting: {cur_grad_scale}"
|
||||||
)
|
)
|
||||||
|
|
||||||
if params.batch_idx_train > 4000 and loss > 300 and params.wandb:
|
#if params.batch_idx_train > 4000 and loss > 300 and params.wandb:
|
||||||
wb.log({"valid/loss": 10000})
|
# wb.log({"valid/loss": 10000})
|
||||||
raise RuntimeError(
|
# raise RuntimeError(
|
||||||
f"divergence... exiting: loss={loss}"
|
# f"divergence... exiting: loss={loss}"
|
||||||
)
|
# )
|
||||||
|
|
||||||
if batch_idx % (params.log_interval*params.accum_grads) == 0:
|
if batch_idx % (params.log_interval*params.accum_grads) == 0:
|
||||||
#for n, p in model.named_parameters():
|
#for n, p in model.named_parameters():
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user