如果你用Pytorch定義的網絡結構太過于靈活，那么轉成ONNX的時候很有可能出錯。這個報錯通常情況下是你的網絡結構中出現if else 語句。比如

if cfg.CUDA:eps = torch.cuda.FloatTensor(std.size()).normal_()else:eps = torch.FloatTensor(std.size()).normal_()

最好不要這樣寫，能避免的話盡量避免。不要寫if else判斷語句。
比如寫成這樣：

eps = torch.normal(torch.zeros_like(mu))

如果避免不了的話，(⊙o⊙)…，可能轉出來的網絡就有問題。

參考資料

Pytorch 1.0 Tracer Warning: Converting a tensor to a Python index might cause the trace to be incorrect

總結

以上是生活随笔為你收集整理的Pytorch转ONNX采坑记：Converting a tensor to a Python boolean might cause the trace to be incorrect. We...的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。