merge

pcsl-epfl · Aug 24, 2023 · d63897d · d63897d
2 parents 1645e80 + a9e3d90
commit d63897d
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 179 deletions.
diff --git a/models/cnn2.py b/models/cnn2.py
diff --git a/models/fcn.py b/models/fcn.py
@@ -31,7 +31,7 @@ def forward(self, x):
 
 class FCN(nn.Module):
     def __init__(self, input_channels, h, out_dim, num_layers, bias=False):
-        super(FCN2, self).__init__()
+        super(FCN, self).__init__()
 
         d = 2 ** num_layers
 

diff --git a/models/rf.py b/models/rf.py