2020年8月20日
pytorch转ONNX以及TnesorRT的坑
摘要： 1、upsample在使用的时候，比如插值，F.in.. 是scale设置为具体的尺寸，不能是比例 2、使用sclice的时候，只能是H和W维度，C维度用split 阅读全文
posted @ 2020-08-20 16:46 阿刚的代码进阶之旅 阅读(374) 评论(0) 推荐(0) 编辑