How to use weights of Tensorflow model in pytorch?

There’s no tricks to it - you just have to dump the weights and reload them. It’s quite time consuming! Lots have already been done here BTW: https://github.com/Cadene/pretrained-models.pytorch/

1 Like