diff --git a/fluid/PaddleNLP/neural_machine_translation/transformer/train.py b/fluid/PaddleNLP/neural_machine_translation/transformer/train.py index 7a816ad05b77ea51054cf2d73c282b51595ec31a..5fc98868aa6e36bc5d1c5c0ad7ab231cda0fd52d 100644 --- a/fluid/PaddleNLP/neural_machine_translation/transformer/train.py +++ b/fluid/PaddleNLP/neural_machine_translation/transformer/train.py @@ -496,8 +496,7 @@ def train_loop(exe, np.log(TrainTaskConfig.label_smooth_eps / ( ModelHyperParams.trg_vocab_size - 1) + 1e-20)) - # num_iteration_per_drop_scope start from 1 - step_idx = 1 + step_idx = 0 init_flag = True logging.info("begin train")