Update model2_cpx.py

Linfang-mumu · Jun 20, 2021 · aa89098 · aa89098
1 parent bf608bd
commit aa89098
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/model2_cpx.py b/model2_cpx.py
@@ -105,7 +105,7 @@ def __init__(
 
         super(ResBlock, self).__init__()
         m = []
-        for i in range(2):
+        for i in range(3):
             m.append(ComplexConv2d(n_feats, n_feats, kernel_size, bias=bias))
             if bn:
                 m.append(nn.BatchNorm2d(n_feats))
@@ -187,7 +187,7 @@ def forward(self, x):
 #################################################
 
         y = x
-        for i in range(2):
+        for i in range(3):
             x = y
             new_k = torch.complex(x[:, 0, :, :], x[:, 1, :, :])
 
@@ -668,7 +668,7 @@ def forward(self, x):
 #################################################
 
         y = x
-        for i in range(2):
+        for i in range(3):
             x = y
             new_k = torch.complex(x[:, 0, :, :], x[:, 1, :, :])