scglue.models.plugins.EarlyStopping.attach

EarlyStopping.attach(net, trainer, train_engine, val_engine, train_loader, val_loader, directory)

Attach custom handlers to training or validation engine

参数:

• net (Module) – Network module

• trainer (scglue.models.base.Trainer) – Trainer object

• train_engine (ignite.engine.engine.Engine) – Training engine

• val_engine (ignite.engine.engine.Engine) – Validation engine

• train_loader (typing.Iterable) – Training data loader

• val_loader (typing.Optional[typing.Iterable]) – Validation data loader

• directory (pathlib.Path) – Training directory

返回类型:

None