how to convert the model to onnx？

myungsub / CAIN

Source code for AAAI 2020 paper "Channel Attention Is All You Need for Video Frame Interpolation"

MIT License

320 stars 43 forks source link

how to convert the model to onnx？ #3

Closed hiredd closed 4 years ago

hiredd commented 4 years ago

when i want to convert the model to the onnx , but I can‘’t find the input and output name of the model which needed by torch.onnx.export(), can you help me? how to find the input and the output name? thanks !

myungsub commented 4 years ago

Hi, I have no experience with onnx, but briefly looking at it docs, I think you can just specify input & output names as you want. So, I guess you can just set it to something like

input_names = ["input_1", "input_2"]
output_names = ["output_frame", "output_features"]

hiredd commented 4 years ago

good job! thank you very much ~