Open
Description
rcs3-yolo模型无法导出onnx，报错信息为F.avg_pool2d()的kernel_size参数类型时tensor，不是int型。查询到SEBlock模块中有一个F.avg_pool2d()算子,kernel_size=input_size(3)。
Metadata
Metadata
Assignees
Labels
No labels
{{ message }}
No one assigned
No labels
No projects
No milestone
None yet
No branches or pull requests