scglue.models.base.TrainingPlugin.attach

abstract TrainingPlugin.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 module

  • trainer (Trainer) – Trainer object

  • train_engine (Engine) – Training engine

  • val_engine (Engine) – Validation engine

  • train_loader (Iterable) – Training data loader

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

  • directory (Path) – Training directory

Return type

None