Basic Framework

modules/pipeline.py

@@ -0,0 +1,20 @@
+from torch import nn
+from configs.config import ConfigManager
+
+
+class Pipeline(nn.Module):
+    TRAIN_MODE: str = "train"
+    TEST_MODE: str = "test"
+
+    def __init__(self, pipeline_config):
+        super(Pipeline, self).__init__()
+
+        self.modules_config = ConfigManager.get("modules")
+        self.device = ConfigManager.get("settings", "general", "device")
+
+    def forward(self, data, mode):
+        pass
+
+
+if __name__ == '__main__':
+    pass