serotiny.models.base_model module# class serotiny.models.base_model.BaseModel(**kwargs)[source]# Bases: LightningModule configure_optimizers()[source]# forward(x, **kwargs)[source]# parse_batch(batch)[source]# predict_step(batch, batch_idx)[source]# test_epoch_end(outputs)[source]# test_step(batch, batch_idx)[source]# train_epoch_end(outputs)[source]# training_step(batch, batch_idx, optimizer_idx=None)[source]# validation_epoch_end(outputs)[source]# validation_step(batch, batch_idx)[source]#