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/modules/utils.py
+++ b/modules/utils.py
@ -11,8 +11,16 @@ import pytorch_lightning as pl

 # Utility - Modules
 ###################
+class F_x(object):
+    def __init__(self):
+        pass
+
+    def __call__(self, x):
+        return x


+# Utility - Modules
+###################
 class Flatten(nn.Module):

    @property