LinearModule

2020-05-09 21:56:57 +02:00
parent d2e74ff33a
commit f6c6726509
3 changed files with 51 additions and 36 deletions
--- a/utils/model_io.py
+++ b/utils/model_io.py
@ -18,19 +18,12 @@ class ModelParameters(Namespace, Mapping):

        paramter_mapping.update(
            dict(
-                activation=self._activations[paramter_mapping['activation']]
+                activation=self._activations[self['activation']]
            )
        )

        return paramter_mapping

-    @property
-    def test_activation(self):
-        try:
-            return self._activations[self.model.activation]
-        except KeyError:
-            return nn.ReLU
-
    def __getitem__(self, k):
        # k: _KT -> _VT_co
        return self.__dict__[k]