pytorch-onnx-tensorflow-pb icon indicating copy to clipboard operation
pytorch-onnx-tensorflow-pb copied to clipboard

Converting A PyTorch Model to Tensorflow pb using ONNX

Results 3 pytorch-onnx-tensorflow-pb issues
Sort by recently updated
recently updated
newest added

onnx转pb模型的时候,报这个错误: Traceback (most recent call last): File "onnx2pb.py", line 45, in onnx2pb_2(onnx_input_path) File "onnx2pb.py", line 14, in onnx2pb_2 tf_rep = prepare(model) File "/home/fffan/下载/onnx-tensorflow-rel-1.6.0/onnx_tf/backend.py", line 66, in prepare return cls.onnx_model_to_tensorflow_rep(model, strict)...

你好!我把pytorch模型转换成onnx后,用同一个输入,两个模型输出的结果不一致该怎么办呢?torch.onnx.export中的参数model应该是没有问题的,会是因为其他参数吗?

@cinastanbean thanks for your great job,but i can't find the model_best_checkpoint_resnet18.pth.tar ,it's different with [it](https://github.com/pytorch/vision/blob/master/torchvision/models/resnet.py),can you post this weights,looking for your replay,thanks very much.[](url)