search

rxn4chemistry / rxnfp

Reaction fingerprints, atlases and classification. Code complementing our Nature Machine Intelligence publication on "Mapping the space of chemical reactions using attention-based neural networks" (http://rdcu.be/cenmd).
https://rxn4chemistry.github.io/rxnfp/
MIT License
156 stars 40 forks source link

during the import transformer no SAVE_STATE_WARNING #7

Closed qiaohaoran closed 3 years ago

qiaohaoran commented 3 years ago

image I've commented it out，and the example have good run。 it is ok?

pschwllr commented 3 years ago

This due to a recent change in torch 1.8.0. If you downgrade to 

pip install torch==1.7.1

the import will work.

pschwllr commented 3 years ago

I've updated the pypi package (0.0.9), it has now torch>=1.3.1,<1.8 as dependency.

qiaohaoran commented 3 years ago

thank you very much ,awesome