diff --git a/module/resnet_block.py b/module/resnet_block.py index ca904ab..59f1c42 100644 --- a/module/resnet_block.py +++ b/module/resnet_block.py @@ -142,7 +142,7 @@ def forward(self, x): torch.Tensor: Output tensor after residual connection """ # Residual connection: output = input + conv_block(input) - out = x + self.conv_block(x) + out = x + self.conv_block(x) #残差连接 return out # def forward(self, x):