TensorRT:Scale Layer

運行googlenet,遇到的問題:Pytorch和C++版本RT的結果不一致。 pytorch結果: C++版本RT結果: 按理說應該是一樣的,怎麼會不一樣呢? 肯定是自己的問題,所以查一下問題所在。   其實問題就出現在這行代碼上: 如果pretrained=True,torchvision.models不僅會創建網絡,load權重,還會做圖像的預處理(這個我以前竟然不知道。。) 如果pret
相關文章
相關標籤/搜索