search

kkahatapitiya / X3D-Multigrid

PyTorch implementation of X3D models with Multigrid training.
MIT License
92 stars 13 forks source link

Model convertion #1

Closed OValery16 closed 3 years ago

OValery16 commented 3 years ago

Thank you a lot for sharing your implementation. It is really helpful for implementing X3D network on custom deep learning problem.

The original repo only provides Caffe2 pretrained model. How did you convert them to pytorch format ? (I also want to try other version of X3D)

kkahatapitiya commented 3 years ago

I followed a brute force approach of mapping parameter names from their version to my version. It may feel like a lot of work at first, but with print(model) and copy/paste it doesn't take that long once you see the pattern in each block.

OValery16 commented 3 years ago

Thank you for your help.