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

updated the assert to check if the seq_length == 196_608 when use_tf_… #41

Closed jerome-f closed 8 months ago

jerome-f commented 8 months ago

…gamma=True

Update modeling_enformer.py to regular behavior of seq_length==196608.

I can verify that post this both TF-sonnet and the pytorch predictions are almost exact.