Debugging

lib/utils/config.py

@@ -5,6 +5,7 @@ from collections import defaultdict
 from configparser import ConfigParser
 from pathlib import Path
 
+from lib.models.homotopy_classification.cnn_based import ConvHomDetector
 from lib.utils.model_io import ModelParameters
 
 
@@ -24,6 +25,15 @@ class Config(ConfigParser):
     # for section in self.default_sections:
     #     self.__setattr__(section, property(lambda x :x._get_namespace_for_section(section))
 
+    @property
+    def model_class(self):
+        model_dict = dict(classifier_cnn=ConvHomDetector)
+        try:
+            return model_dict[self.get('model', 'type')]
+        except KeyError as e:
+            raise KeyError(rf'The model alias you provided ("{self.get("model", "type")}") does not exist! \n'
+                           f'Try one of these:\n{list(model_dict.keys())}')
+
     @property
     def main(self):
         return self._get_namespace_for_section('main')

lib/utils/logging.py

@@ -16,6 +16,12 @@ class Logger(LightningLoggerBase):
         else:
             return self.neptunelogger.experiment
 
+    @property
+    def log_dir(self):
+        if self.debug:
+            return Path(self.outpath)
+        return Path(self.experiment.log_dir).parent
+
     @property
     def name(self):
         return self.config.model.type
@@ -50,10 +56,9 @@ class Logger(LightningLoggerBase):
         self.debug = debug
         self.config = config
         self._testtube_kwargs = dict(save_dir=self.outpath, version=self.version, name=self.name)
-        self._neptune_kwargs = dict(offline_mode= self.debug,
+        self._neptune_kwargs = dict(offline_mode=self.debug,
                                     api_key=self.config.project.neptune_key,
                                     project_name=self.project_name,
-                                    name=self.name,
                                     upload_source_files=list())
         self.neptunelogger = NeptuneLogger(**self._neptune_kwargs)
         self.testtubelogger = TestTubeLogger(**self._testtube_kwargs)
@@ -67,3 +72,6 @@ class Logger(LightningLoggerBase):
         self.neptunelogger.log_metrics(metrics, step_num)
         self.testtubelogger.log_metrics(metrics, step_num)
         pass
+
+    def log_config_as_ini(self):
+        self.config.write(self.log_dir)