from local
This commit is contained in:
parent
871b84fedd
commit
c6c8facaa5
Binary file not shown.
@ -955,8 +955,6 @@ def train_one_epoch(
|
|||||||
)
|
)
|
||||||
loss_info.reduce(loss.device)
|
loss_info.reduce(loss.device)
|
||||||
|
|
||||||
logging.info(loss_info)
|
|
||||||
|
|
||||||
numel = params.world_size / (params.accum_grads * loss_info["utterances"])
|
numel = params.world_size / (params.accum_grads * loss_info["utterances"])
|
||||||
loss *= numel ## normalize loss over utts(batch size)
|
loss *= numel ## normalize loss over utts(batch size)
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user