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)[source]#

Supervised model.

Parameters:

  • model (

    Default:) – Config to build a model.

  • optimizer (

    Default:) – Config to build optimizers and schedulers.

  • train_transform (

    Default:, optional) – Config to perform transformation on train input.
    Default: None

  • val_transform (

    Default:, optional) – Config to perform transformation on val input.
    Default: None

  • test_transform (

    Default:, optional) – Config to perform transformation on test input.
    Default: None

  • val_time_augmentation (

    Default:, optional) – Ensembling method for test time augmentation used at validation.
    Default: None

  • test_time_augmentation (

    Default:, optional) – Ensembling method for test time augmentation used at test.
    Default: None