search

flatironinstitute / DeepFRI

Deep functional residue identification
BSD 3-Clause "New" or "Revised" License
305 stars 78 forks source link

About train details #29

Open TengQinglong opened 1 year ago

TengQinglong commented 1 year ago

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 commented 1 year ago

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.