Hparams passing with user warnings

2020-04-17 18:07:54 +02:00
parent e53107420d
commit f5c240f038
5 changed files with 27 additions and 16 deletions
--- a/utils/config.py
+++ b/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