pytorch转ONNX以及TnesorRT的坑

1、upsample在使用的时候，比如插值，F.in..

是scale设置为具体的尺寸，不能是比例

2、使用sclice的时候，只能是H和W维度，C维度用split

posted @ 2020-08-20 16:46  阿刚的代码进阶之旅  阅读(372)  评论(0)  编辑  收藏  举报