Browse Source

little change

tags/v1.0.0alpha
YWMditto 3 years ago
parent
commit
a61f3975d7
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      fastNLP/core/controllers/trainer.py

+ 3
- 3
fastNLP/core/controllers/trainer.py View File

@@ -174,13 +174,13 @@ class Trainer(TrainerEventTrigger):
optimizers=optimizers, optimizers=optimizers,
device=device, device=device,
n_epochs=n_epochs, n_epochs=n_epochs,
validate_dataloaders=evaluate_dataloaders,
evaluate_dataloaders=evaluate_dataloaders,
batch_step_fn=batch_step_fn, 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, evaluate_fn=evaluate_fn,
callbacks=callbacks, callbacks=callbacks,
metrics=metrics, metrics=metrics,
validate_every=evaluate_every,
evaluate_every=evaluate_every,
input_mapping=input_mapping, input_mapping=input_mapping,
output_mapping=output_mapping, output_mapping=output_mapping,
model_wo_auto_param_call=model_wo_auto_param_call, model_wo_auto_param_call=model_wo_auto_param_call,


Loading…
Cancel
Save