Hparams passing with user warnings

utils/config.py

@@ -116,7 +116,7 @@ class Config(ConfigParser, ABC):
     def build_model(self):
         return self.model_class(self.model_paramters)
 
-    def build_and_init_model(self, weight_init_function):
+    def build_and_init_model(self, in_shape, weight_init_function):
         model = self.build_model()
         model.init_weights(weight_init_function)
         return model

utils/logging.py

@@ -37,8 +37,8 @@ class Logger(LightningLoggerBase):
 
     @property
     def outpath(self):
-        # ToDo: Add further path modification such as dataset config etc.
-        return Path(self.config.train.outpath) / self.config.data.mode
+        # FIXME: Move this out of here, this is not the right place to do this!!!
+        return Path(self.config.train.outpath) / self.config.model.type
 
     def __init__(self, config: Config):
         """