onnx / onnx-tensorflow

Tensorflow Backend for ONNX
Other
1.29k stars 295 forks source link

PADing error on resnet model from Pytorch #188

Open inakinavarro opened 6 years ago

inakinavarro commented 6 years ago

When converting a model ResNet based model created with Pytorch I get the following warning message:

_onnx_tf/backends/backendv1.py:123: UserWarning: Using the pooling op in compatibility mode.This means your graph cannot be serialized.Please configure your pooling operation to only use paddings that correspond to Tensorflow SAME or VALID padding. "correspond to Tensorflow SAME or VALID padding.", UserWarning)

I use the following code to convert: python onnx_2_tf.py hope.onnx

import onnx_tf.backend as tf_backend
import onnx
import argparse

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('model', help='onnx file')
    args = parser.parse_args()

    model = onnx.load(args.model)
    tf_backend.prepare(model)

if __name__ == "__main__":
    main()

Here you can find the model: https://www.dropbox.com/s/nj2xceg14ctwdxr/hope.onnx?dl=0

Python, ONNX, ONNX-TF, Tensorflow version

This section can be obtained by running get_version.py from util folder.

fumihwh commented 6 years ago

Maybe duplicate of #167