Oneflow-Inc / OneFlow-Benchmark

OneFlow models for benchmarking.
104 stars 31 forks source link

fix resnet2onnx bug #177

Closed MARD1NO closed 3 years ago

MARD1NO commented 3 years ago

给resnet2onnx.py增加args参数,并配有shell脚本

由于预训练模型没有做bn relu融合,所以在shell脚本里,相关选项置为False,目前已成功运行正确结果

图片