lucidrains/enformer-pytorch

enformer TF pretrained weights

naumanjaved opened this issue · 1 comments

Hello!

Thanks for this wonderful resource. I was wondering whether you can point me to how to obtain the model weights for the original TF version of Enformer, or the actual weights if they are stored somewhere easily accessible. I see the model on TF hub but am not sure exactly how to extract the weights - I seem to be running into some issues potentially because the original code is sonnet based and the model is always loaded as a custom user object..

Much appreciated!

sorry - found them located at :

gs://dm-enformer/models/enformer/sonnet_weights/