steffen/hom_traj_gen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

train running dataset fixed

Browse Source
This commit is contained in:
steffen
2020-03-05 20:50:07 +01:00
parent 1f25bf599b
commit 05033bed75
11 changed files with 41 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lib/utils/logging.py
View File

@ -76,7 +76,7 @@ class Logger(LightningLoggerBase):
self.neptunelogger.close()
def log_config_as_ini(self):
self.config.write(self.log_dir)
self.config.write(self.log_dir / 'config.ini')
def log_metric(self, metric_name, metric_value, **kwargs):
self.testtubelogger.log_metrics(dict(metric_name=metric_value))
@ -91,8 +91,8 @@ class Logger(LightningLoggerBase):
self.neptunelogger.save()
def finalize(self, status):
self.testtubelogger.finalize()
self.neptunelogger.finalize()
self.testtubelogger.finalize(status)
self.neptunelogger.finalize(status)
self.log_config_as_ini()
def __enter__(self):