scglue.models.plugins.EarlyStopping.attach
- EarlyStopping.attach(net, trainer, train_engine, val_engine, train_loader, val_loader, directory)[source]
Attach custom handlers to training or validation engine
- Parameters
net (
Module
) – Network moduletrainer (
scglue.models.base.Trainer
) – Trainer objecttrain_engine (
ignite.engine.engine.Engine
) – Training engineval_engine (
ignite.engine.engine.Engine
) – Validation enginetrain_loader (
typing.Iterable
) – Training data loaderval_loader (
typing.Optional
[typing.Iterable
]) – Validation data loaderdirectory (
pathlib.Path
) – Training directory
- Return type