pytorch转ONNX以及TnesorRT的坑

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

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

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

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