scglue.models.plugins.Tensorboard.attach

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

Attach custom handlers to training or validation engine

Parameters:

• 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

Return type:

None