pull/1116/head
weishengyu 2021-08-08 13:14:30 +08:00
parent 7e1aa2d358
commit 3b62e4d401
1 changed files with 1 additions and 1 deletions

View File

@ -85,7 +85,7 @@ class ConvBlock(TheseusLayer):
self.relu = nn.ReLU()
def forward(self, inputs, res_dict=None):
super(ConvBlock, self).forward(inputs, res_dict)
super(ConvBlock, self).forward(inputs, res_dict=res_dict)
x = self.conv1(inputs)
x = self.relu(x)
if self.groups == 2 or self.groups == 3 or self.groups == 4: