BorgwardtLab / proteinshake

Protein structure datasets for machine learning.
https://proteinshake.ai
BSD 3-Clause "New" or "Revised" License
101 stars 9 forks source link

add FLOPs and parameter metrics to task #245

Open timkucera opened 1 year ago

timkucera commented 1 year ago

this is framework dependent, but we can provide one interface for each of the supported frameworks.

Would look something like: metrics = task.profile(my_model)