logging active and clean

.idea/deployment.xml

@@ -1,15 +1,8 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
-  <component name="PublishConfigData" autoUpload="On explicit save action" serverName="traj_gen-AiMachine">
+  <component name="PublishConfigData" autoUpload="On explicit save action" serverName="steffen@aimachine:22">
     <serverData>
-      <paths name="ErLoWa-AiMachine">
-        <serverdata>
-          <mappings>
-            <mapping local="$PROJECT_DIR$" web="/" />
-          </mappings>
-        </serverdata>
-      </paths>
-      <paths name="traj_gen-AiMachine">
+      <paths name="steffen@aimachine:22">
         <serverdata>
           <mappings>
             <mapping deploy="/" local="$PROJECT_DIR$" web="/" />

.idea/hom_traj_gen.iml

@@ -2,7 +2,7 @@
 <module type="PYTHON_MODULE" version="4">
   <component name="NewModuleRootManager">
     <content url="file://$MODULE_DIR$" />
-    <orderEntry type="jdk" jdkName="traj_gen@AiMachine" jdkType="Python SDK" />
+    <orderEntry type="jdk" jdkName="Remote Python 3.7.6 (sftp://steffen@aimachine:22/home/steffen/envs/traj_gen/bin/python)" jdkType="Python SDK" />
     <orderEntry type="sourceFolder" forTests="false" />
   </component>
 </module>

.idea/misc.xml

@@ -3,5 +3,5 @@
   <component name="JavaScriptSettings">
     <option name="languageLevel" value="ES6" />
   </component>
-  <component name="ProjectRootManager" version="2" project-jdk-name="traj_gen@AiMachine" project-jdk-type="Python SDK" />
+  <component name="ProjectRootManager" version="2" project-jdk-name="hom_traj_gen@aimachine" project-jdk-type="Python SDK" />
 </project>

lib/utils/logging.py

@@ -18,9 +18,7 @@ class Logger(LightningLoggerBase):
 
     @property
     def log_dir(self):
-        if self.debug:
-            return Path(self.outpath)
-        return Path(self.experiment.log_dir).parent
+        return Path(self.testtubelogger.experiment.get_logdir()).parent
 
     @property
     def name(self):
@@ -32,14 +30,14 @@ class Logger(LightningLoggerBase):
 
     @property
     def version(self):
-        return f"version_{self.config.get('main', 'seed')}"
+        return self.config.get('main', 'seed')
 
     @property
     def outpath(self):
         # ToDo: Add further path modification such as dataset config etc.
         return Path(self.config.train.outpath)
 
-    def __init__(self, config: Config, debug=False):
+    def __init__(self, config: Config):
         """
         params (dict|None): Optional. Parameters of the experiment. After experiment creation params are read-only.
            Parameters are displayed in the experiment’s Parameters section and each key-value pair can be
@@ -53,8 +51,8 @@ class Logger(LightningLoggerBase):
         """
         super(Logger, self).__init__()
 
-        self.debug = debug
         self.config = config
+        self.debug = self.config.main.debug
         self._testtube_kwargs = dict(save_dir=self.outpath, version=self.version, name=self.name)
         self._neptune_kwargs = dict(offline_mode=self.debug,
                                     api_key=self.config.project.neptune_key,
@@ -68,9 +66,9 @@ class Logger(LightningLoggerBase):
         self.testtubelogger.log_hyperparams(params)
         pass
 
-    def log_metrics(self, metrics, step_num):
-        self.neptunelogger.log_metrics(metrics, step_num)
-        self.testtubelogger.log_metrics(metrics, step_num)
+    def log_metrics(self, metrics, step=None):
+        self.neptunelogger.log_metrics(metrics, step=step)
+        self.testtubelogger.log_metrics(metrics, step=step)
         pass
 
     def log_config_as_ini(self):

main.py

@@ -66,7 +66,7 @@ config = Config.read_namespace(args)
 
 # Logger
 # =============================================================================
-logger = Logger(config, debug=True)
+logger = Logger(config)
 
 # Checkpoint Callback
 # =============================================================================
@@ -95,7 +95,7 @@ if __name__ == "__main__":
                       log_save_interval=model.data_len // 10,  # TODO: Better Value / Setting
                       checkpoint_callback=checkpoint_callback,
                       logger=logger,
-                      fast_dev_run=config.get('main', 'debug'),
+                      fast_dev_run=config.main.debug,
                       early_stop_callback=None
                       )