Supervised

class eztorch.models.supervised.SupervisedModel(model, optimizer, train_transform=None, val_transform=None, test_transform=None, val_time_augmentation=None, test_time_augmentation=None)

Supervised model.

Parameters:

• model (DictConfig) – Config to build a model.

• optimizer (DictConfig) – Config to build optimizers and schedulers.

• train_transform (Optional[DictConfig], optional) – Config to perform transformation on train input.

  Default: None

• val_transform (Optional[DictConfig], optional) – Config to perform transformation on val input.

  Default: None

• test_transform (Optional[DictConfig], optional) – Config to perform transformation on test input.

  Default: None

• val_time_augmentation (Optional[DictConfig], optional) – Ensembling method for test time augmentation used at validation.

  Default: None

• test_time_augmentation (Optional[DictConfig], optional) – Ensembling method for test time augmentation used at test.

  Default: None