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
b9e221a0c0
commit
f66f6c7d9d
Binary file not shown.
Binary file not shown.
@ -1009,7 +1009,7 @@ def train_one_epoch(
|
|||||||
|
|
||||||
# NOTE: We use reduction==sum and loss is computed over utterances
|
# NOTE: We use reduction==sum and loss is computed over utterances
|
||||||
# in the batch and there is no normalization to it so far.
|
# in the batch and there is no normalization to it so far.
|
||||||
logging.warning(f"Grad scale is small: {cur_grad_scale}")
|
scaler.scale(loss).backward()
|
||||||
|
|
||||||
if params.multi_optim and batch_idx % params.accum_grads == 0:
|
if params.multi_optim and batch_idx % params.accum_grads == 0:
|
||||||
set_batch_count(model, params.batch_idx_train)
|
set_batch_count(model, params.batch_idx_train)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user