def __init__(self,
num_classes=1,
img_sz=128,
model_train_loss=True,
bs: int = 1,
steps_per_epoch: int = 0,
lr: float = 1e-2,
noisy=False,
calc_metrics=False,
**kwargs):
LightningModule.__init__(self)
self.num_classes = num_classes
self.model_train_loss = model_train_loss
self.img_sz = img_sz
self.lr = lr
self.bs = bs
self.steps_per_epoch = steps_per_epoch
self.noisy = noisy
self.calc_metrics = calc_metrics
self.model = self.create_model(num_classes=num_classes,
img_sz=img_sz,
lr=lr,
bs=bs,
steps_per_epoch=steps_per_epoch,
**kwargs)
def __init__(self):
LightningModule.__init__(self)