search

DeepGraphLearning / GearNet

GearNet and Geometric Pretraining Methods for Protein Structure Representation Learning, ICLR'2023 (https://arxiv.org/abs/2203.06125)
MIT License
263 stars 27 forks source link

RuntimeError: addmm: Argument #3 (dense), for training ESM_GearNet on EC #47

Closed hxu105 closed 1 year ago

hxu105 commented 1 year ago

Hi, thank you for sharing the project. I am trying to reproduce the result of the ESM_GearNet model, and I have some problems on fine-tuning it on EC downstream task. Here is the picture.

1692110063132

I was able to pretrain the model on AlphaFoldDB but failed to fine-tune it or train it from scratch on EC.