- 博客(1)
- 资源 (1)
- 收藏
- 关注
RSS订阅原创 Pytorch转ONNX采坑记：Converting a tensor to a Python boolean might cause the trace to be incorrect. We...
如果你用Pytorch定义的网络结构太过于灵活，那么转成ONNX的时候很有可能出错。这个报错通常情况下是你的网络结构中出现if else 语句。比如 if cfg.CUDA: eps = torch.cuda.FloatTensor(std.size()).normal_() else: eps = torch.Flo...
2020-05-02 09:49:29
13277
1
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人