How many paramters are there in ProteinMPNNmodels?

dauparas / ProteinMPNN

Code for the ProteinMPNN paper

MIT License

934 stars 284 forks source link

How many paramters are there in ProteinMPNNmodels? #48

Closed Huanle closed 1 year ago

Huanle commented 1 year ago

Hi @dauparas ,

This is a layman's question inspired by news related to chatGPT and GPT4. There might be a way to count it but i do not know how.

Can you please let me know? Thanks a lot.

dauparas commented 1 year ago

You can run something like pytorch_total_params = sum(p.numel() for p in model.parameters()). The ProteinMPNN model has about 1.66 million parameters.