from lib.modules.utils import LightningBaseModule
from lib.modules.blocks import ConvModule

class ConvHomDetector(LightningBaseModule):

    name = 'CNNHomotopyClassifier'

    def configure_optimizers(self):
        pass

    def validation_step(self, *args, **kwargs):
        pass

    def validation_end(self, outputs):
        pass

    def training_step(self, batch_xy, batch_nb, *args, **kwargs):
        pass

    def test_step(self, *args, **kwargs):
        pass

    def __init__(self, *params):
        super(ConvHomDetector, self).__init__(*params)

        self.conv1 = ConvModule(self.dataset.map_shape

        )


    def forward(self, x):
        pass