Skip to content

O
oneflow_convert_tools

Oneflow-Inc / oneflow_convert_tools
上一次同步 2 年多

通知 1
Star 22
Fork 0
O
oneflow_convert_tools

前往新版Gitcode，体验更适合开发者的 AI 搜索 >>

提交 d4352a5e 编写于 作者: B BBuf
浏览文件
操作

support repvgg

上级 fbfe9d89
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
examples/oneflow2onnx/models/test_repvgg.py
浏览文件 @ d4352a5e
...@@ -190,7 +190,7 @@ class RepVGG(nn.Module): ...@@ -190,7 +190,7 @@ class RepVGG(nn.Module):
self.stage4 = self._make_stage( self.stage4 = self._make_stage(
int(512 * width_multiplier[3]), num_blocks[3], stride=2 int(512 * width_multiplier[3]), num_blocks[3], stride=2
) )
self.gap = nn.AdaptiveAvgPool2d(output_size=1) self.gap = nn.AvgPool2d(7)
self.linear = nn.Linear(int(512 * width_multiplier[3]), num_classes) self.linear = nn.Linear(int(512 * width_multiplier[3]), num_classes)
def _make_stage(self, planes, num_blocks, stride): def _make_stage(self, planes, num_blocks, stride):
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或