search

xxradon / PytorchToCaffe

Pytorch model to caffe model, supported pytorch 0.3, 0.3.1, 0.4, 0.4.1 ,1.0 , 1.0.1 , 1.2 ,1.3 .notice that only pytorch 1.1 have some bugs
MIT License
783 stars 224 forks source link

NET_INITTED=True之后input_var的data属性就变成“Unable to get repr for <class 'torch.Tensor'>” #52

Closed mmchen88 closed 4 years ago

mmchen88 commented 4 years ago

pytorch_to_caffe.py文件里trans_net函数的“global NET_INITTED NET_INITTED=True”这两句执行完了之后，input_var的data属性就变成“Unable to get repr for <class 'torch.Tensor'>”，forward运行不通过。 请问这是为什么呢？

mmchen88 commented 4 years ago

少了 net.eval()