I have init_net.pb and predict_net.pb files of a caffe2 model. I have created the same network in pytorch. Now, I want to copy all the weights from caffe2 to pytorch model. Can someone please guide me how to do that?
These are the layers with trainable weights in caffe2:
conv
spatial_bn
conv_transpose
@ptrblck please help