About train details

[TengQinglong]

Hello, I'm trying to train the model in the same dataset. Could you provide more details about training the model. I'm trying to train your model on the same dataset for task named "mf", but the model will converge in one epoch, and the final model weights doesn't perform as well as the trained models you provided.

[PawelSzczerbiak]

Did you use the PDB structures only or PDB + SWISS-MODEL ones? For the latter you should get an additional boost in performance but the SWISS-MODEL data are not a part of the repo.