Mariewelt / OpenChem

OpenChem: Deep Learning toolkit for Computational Chemistry and Drug Design Research
https://mariewelt.github.io/OpenChem/
MIT License
681 stars 114 forks source link

feat: Allow flexibility in load_state_dict #29

Open ngchanway opened 1 year ago

ngchanway commented 1 year ago

To allow loading state_dict with additional parameters