提交 8211880d 编写于 作者: W weishengyu

dbg

上级 132ec21f
...@@ -140,11 +140,11 @@ class VGGNet(TheseusLayer): ...@@ -140,11 +140,11 @@ class VGGNet(TheseusLayer):
self._update_res(return_patterns) self._update_res(return_patterns)
def forward(self, inputs, res_dict=None): def forward(self, inputs, res_dict=None):
x = self.conv_block_1(inputs, res_dict) x = self.conv_block_1(inputs)
x = self.conv_block_2(x, res_dict) x = self.conv_block_2(x)
x = self.conv_block_3(x, res_dict) x = self.conv_block_3(x)
x = self.conv_block_4(x, res_dict) x = self.conv_block_4(x)
x = self.conv_block_5(x, res_dict) x = self.conv_block_5(x)
x = self.flatten(x) x = self.flatten(x)
x = self.fc1(x) x = self.fc1(x)
x = self.relu(x) x = self.relu(x)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册