project Refactor, CNN Classifier Basics

lib/models/homotopy_classification/cnn_based.py

@@ -0,0 +1,32 @@
+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