From be83411e257da39da46d463d9fb9e4ba81560528 Mon Sep 17 00:00:00 2001
From: Alex Moreo <alejandro.moreo@isti.cnr.it>
Date: Thu, 9 Jul 2020 22:02:48 +0200
Subject: [PATCH] bugfix

---
 src/model/layers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/model/layers.py b/src/model/layers.py
index 66ba056..3e2b87d 100644
--- a/src/model/layers.py
+++ b/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