bugfix

src/model/layers.py

@@ -6,7 +6,7 @@ import torch.nn.functional as F
 
 class Phi(nn.Module):
 
-    def __init__(self, cnn, ff, norm):
+    def __init__(self, cnn, ff, norm=None):
         super(Phi, self).__init__()
         self.cnn = cnn
         self.ff = ff