Model blocks, Model files, rearrange project structure

lib/models/cnn.py

@@ -0,0 +1,29 @@
+from lib.models.blocks import RecurrentModule, ConvModule, DeConvModule, Generator, LightningBaseModule
+
+
+class CNNRouteGeneratorModel(LightningBaseModule):
+
+    @classmethod
+    def name(cls):
+        pass
+
+    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(CNNRouteGeneratorModel, self).__init__(*params)
+
+    def forward(self, x):
+        pass