From 3b62e4d40121a3becaa7d173755bca4e4ebe3dec Mon Sep 17 00:00:00 2001 From: weishengyu Date: Sun, 8 Aug 2021 13:14:30 +0800 Subject: [PATCH] dbg --- ppcls/arch/backbone/legendary_models/vgg.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppcls/arch/backbone/legendary_models/vgg.py b/ppcls/arch/backbone/legendary_models/vgg.py index 7d88fe9c6..1497aa6f8 100644 --- a/ppcls/arch/backbone/legendary_models/vgg.py +++ b/ppcls/arch/backbone/legendary_models/vgg.py @@ -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: