Update src/model.py

src/model.py

@@ -7,11 +7,11 @@ class Clamp(nn.Module):
 class B(nn.Module):
     def __init__(self, n_in, n_out):
         super().__init__()
-        self.C = nn.Sequential(C(n_in, n_out), nn.ReLU(), C(n_out, n_out), nn.ReLU(), C(n_out, n_out))
+        self.conv = nn.Sequential(C(n_in, n_out), nn.ReLU(), C(n_out, n_out), nn.ReLU(), C(n_out, n_out))
         self.skip = nn.Conv2d(n_in, n_out, 1, bias=False) if n_in != n_out else nn.Identity()
         self.fuse = nn.ReLU()
     def forward(self, x):
-        return self.fuse(self.C(x) + self.skip(x))
+        return self.fuse(self.conv(x) + self.skip(x))
 def E(latent_channels=4):
     return nn.Sequential(
         C(3, 64), B(64, 64),