search

seyonechithrananda / bert-loves-chemistry

bert-loves-chemistry: a repository of HuggingFace models applied on chemical SMILES data for drug design, chemical modelling, etc.
MIT License
389 stars 60 forks source link

only prune state dict if exists #38

Closed ElanaPearl closed 3 years ago

ElanaPearl commented 3 years ago

In the case that we're using a non-local dir, this means pruning won't break anything

Once we upload MTR dataset, this will need to be changed but for now is necessary to finetune on MLM