wklix / TransLNP

MIT License
5 stars 0 forks source link

Questions about released weights #3

Open snpushpi opened 1 month ago

snpushpi commented 1 month ago

Thanks for the great work. i am curious what's the difference between the two model weights that've been released?

https://github.com/wklix/LNPs/releases/download/v1.0/mol_pre_all_h_220816.pt https://github.com/wklix/LNPs/releases/download/v1.0/mol_pre_no_h_220816.pt

I think https://github.com/wklix/LNPs/releases/download/v1.0/mol_pre_no_h_220816.pt gets loaded by default, why is this the case?

wklix commented 1 month ago

The difference between mol_pre_all_h_220816.pt and mol_pre_no_h_220816.pt lies in whether hydrogen atoms were ignored during the pre-training phase. mol_pre_no_h_220816.pt gets loaded by default because remove_hs=True in train_model.py.

At 2024-07-19 06:31:59, "Subha Nawer Pushpita" @.***> wrote:

Thanks for the great work. i am curious what's the difference between the two model weights that've been released?

https://github.com/wklix/LNPs/releases/download/v1.0/mol_pre_all_h_220816.pt https://github.com/wklix/LNPs/releases/download/v1.0/mol_pre_no_h_220816.pt

I think https://github.com/wklix/LNPs/releases/download/v1.0/mol_pre_no_h_220816.pt gets loaded by default, why is this the case?

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you are subscribed to this thread.Message ID: @.***>