fig clf inserted and not resize on kld

lib/utils/config.py

@@ -5,7 +5,9 @@ from collections import defaultdict
 from configparser import ConfigParser
 from pathlib import Path
 
-from lib.models.generators.cnn import CNNRouteGeneratorModel, CNNRouteGeneratorDiscriminated
+from lib.models.generators.cnn import CNNRouteGeneratorModel
+from lib.models.generators.cnn_discriminated import CNNRouteGeneratorDiscriminated
+
 from lib.models.homotopy_classification.cnn_based import ConvHomDetector
 from lib.utils.model_io import ModelParameters
 from lib.utils.transforms import AsArray

lib/utils/logging.py

@@ -37,7 +37,7 @@ class Logger(LightningLoggerBase):
     @property
     def outpath(self):
         # ToDo: Add further path modification such as dataset config etc.
-        return Path(self.config.train.outpath)
+        return Path(self.config.train.outpath) / self.config.data.mode
 
     def __init__(self, config: Config):
         """

lib/utils/tools.py

@@ -9,6 +9,7 @@ def write_to_shelve(file_path, value):
     with shelve.open(str(file_path), protocol=pickle.HIGHEST_PROTOCOL) as f:
         new_key = str(len(f))
         f[new_key] = value
+    f.close()
 
 
 def load_from_shelve(file_path, key):