TensorRT:Scale Layer

运行googlenet,遇到的问题:Pytorch和C++版本RT的结果不一致。 pytorch结果: C++版本RT结果: 按理说应该是一样的,怎么会不一样呢? 肯定是自己的问题,所以查一下问题所在。   其实问题就出现在这行代码上: 如果pretrained=True,torchvision.models不仅会创建网络,load权重,还会做图像的预处理(这个我以前竟然不知道。。) 如果pret
相关文章
相关标签/搜索