from local
This commit is contained in:
parent
f057f0376b
commit
6089ce6b2e
Binary file not shown.
@ -1037,6 +1037,7 @@ def train_one_epoch(
|
|||||||
batch=batch,
|
batch=batch,
|
||||||
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,
|
||||||
|
pl_texts = pl_texts,
|
||||||
)
|
)
|
||||||
loss_info.reduce(loss.device)
|
loss_info.reduce(loss.device)
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user