I have a weight file whose format is hdf5. I want to load the weight parameters in hdf5 into my model.
How can i make it happen in pytorch?
I have a weight file whose format is hdf5. I want to load the weight parameters in hdf5 into my model.
How can i make it happen in pytorch?
Have a look at https://github.com/szagoruyko/functional-zoo/blob/master/resnet-18-export.ipynb
hickle
is actually an interface for hdf5
, so the example is actually loading a hdf5 file (but using the functional interface of pytorch, but the same thing can be done for models defined using nn.Module
)
Could you please tell me how to load weights to hdf5 file.
I am quite new to this.
Thanks !
What if, reversely, I want to export a .h5 file out of an existing CNN constructed with pytorch?
Is there any trick for this please?