Dataset rdy

2021-02-16 10:18:03 +01:00
parent a966321576
commit b5e3e5aec1
3 changed files with 21 additions and 12 deletions
--- a/utils/model_io.py
+++ b/utils/model_io.py
@@ -31,6 +31,8 @@ class ModelParameters(Namespace, Mapping):
                activation=self.__getattribute__('activation')
            )
        )
+        # Get rid of paramters that
+        paramter_mapping.__delitem__('in_shape')

        return paramter_mapping