From aa890989b8c37c9ad588c1ad09eeb9ff002ec8f3 Mon Sep 17 00:00:00 2001
From: Linfang-mumu <52233121+Linfang-mumu@users.noreply.github.com>
Date: Sun, 20 Jun 2021 15:40:17 +0800
Subject: [PATCH] Update model2_cpx.py

---
 model2_cpx.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/model2_cpx.py b/model2_cpx.py
index 328a6d8..c6eb8f9 100644
--- 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, :, :])