diff --git a/fastNLP/core/controllers/trainer.py b/fastNLP/core/controllers/trainer.py index 5154c580..3859fca0 100644 --- a/fastNLP/core/controllers/trainer.py +++ b/fastNLP/core/controllers/trainer.py @@ -174,13 +174,13 @@ class Trainer(TrainerEventTrigger): optimizers=optimizers, device=device, n_epochs=n_epochs, - validate_dataloaders=evaluate_dataloaders, + evaluate_dataloaders=evaluate_dataloaders, batch_step_fn=batch_step_fn, - validate_batch_step_fn=evaluate_batch_step_fn, + evaluate_batch_step_fn=evaluate_batch_step_fn, evaluate_fn=evaluate_fn, callbacks=callbacks, metrics=metrics, - validate_every=evaluate_every, + evaluate_every=evaluate_every, input_mapping=input_mapping, output_mapping=output_mapping, model_wo_auto_param_call=model_wo_auto_param_call,