def __init__(self, cfg):
    '\n        model: torch.nn.Module\n        cfg: model-agnostic experiment configs\n        '
    super().__init__()
    self.cfg = cfg
    self.image = ('MAGNETOGRAM' in cfg.DATA.FEATURES)
    self.model = build_model(cfg)
    self.save_hyperparameters()
model: torch.nn.Module
cfg: model-agnostic experiment configs
def __init__(self, cfg):
    '\n        model: torch.nn.Module\n        cfg: model-agnostic experiment configs\n        '
    super().__init__()
    self.cfg = cfg
    self.image = ('MAGNETOGRAM' in cfg.DATA.FEATURES)
    self.model = build_model(cfg)
    self.save_hyperparameters()