search

longcw / pytorch2caffe

Convert PyTorch model to Caffemodel
541 stars 151 forks source link

How to feed a tuple of inputs? #3

Open ankitmishra262 opened 6 years ago

ankitmishra262 commented 6 years ago

For pytorch2caffe.py, I was wondering whether I can input a tuple of inputs to the input_var? Would it be as simple as commenting this line, because I cannot find where the input_dim is being used again.

If what I am thinking is wrong, how do I feel multiple inputs to the model from pytorch2caffe(input_var, output_var, 'somemodel-pytorch2caffe.prototxt', 'somemodel-pytorch2caffe.caffemodel')?

I am trying to use this for Faster-RCNN.

Thanks in advance.

XavierCHEN34 commented 6 years ago

hello, have you solved this? I just meet the same problem while my pytorch model has two inputs