Viewing a single comment thread. View all comments

parabellum630 t1_ixuno8d wrote

Reply to comment by Deep-Station-1746 in [D] Pytorch or TensorFlow for development and deployment? by CodaholicCorgi

Oh no! I had no idea about ONNX and just transfered a huge model from tf to pyt manually.

18

JustOneAvailableName t1_ixv2bz6 wrote

ONNX tracers are decent, but aren't that good, I had to rewrite Wav2vec to get both batch size and input length as dynamic axis.

8

parabellum630 t1_ixvapdu wrote

I see. But can they be used as a good starting point to build upon?

2

agupta12 t1_ixwfbow wrote

I have been trying to convert wav2vec2 model to onnx without much success

1

CodaholicCorgi OP t1_iy1rgxx wrote

Agreed with all the folks, if your model is far beyond ordinary CNN and RNN then ONNX will give you a bunch of warnings that confuses you.

2