From a61f3975d79a9f391a3661891aa95a444e8be88a Mon Sep 17 00:00:00 2001 From: YWMditto Date: Fri, 15 Apr 2022 00:17:01 +0800 Subject: [PATCH] little change --- fastNLP/core/controllers/trainer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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,