scglue.models.base.Trainer.fit

Trainer.fit(train_loader, val_loader=None, max_epochs=100, random_seed=0, directory=None, plugins=None)

Fit network

Parameters:

• train_loader (typing.Iterable) – Training data loader

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

• max_epochs (int) – Maximal number of epochs

• random_seed (int) – Random seed

• directory (typing.Optional[os.PathLike]) – Training directory

• plugins (typing.Optional[typing.List[scglue.models.base.TrainingPlugin]]) – Optional list of training plugins

Return type:

None