如果你用Pytorch定義的網(wǎng)絡(luò)結(jié)構(gòu)太過于靈活，那么轉(zhuǎn)成ONNX的時(shí)候很有可能出錯(cuò)。這個(gè)報(bào)錯(cuò)通常情況下是你的網(wǎng)絡(luò)結(jié)構(gòu)中出現(xiàn)if else 語(yǔ)句。比如

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

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

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

如果避免不了的話，(⊙o⊙)…，可能轉(zhuǎn)出來(lái)的網(wǎng)絡(luò)就有問題。

參考資料

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

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。