|
|
@@ -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, |
|
|
|