search

lucidrains / enformer-pytorch

Implementation of Enformer, Deepmind's attention network for predicting gene expression, in Pytorch
MIT License
434 stars 81 forks source link

enformer TF pretrained weights #11

Closed naumanjaved closed 2 years ago

naumanjaved commented 2 years ago

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!

naumanjaved commented 2 years ago

sorry - found them located at :

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