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

fixed classification issue; added gitignore #32

Closed walid0925 closed 3 years ago

walid0925 commented 3 years ago

when finetuning, the num_labels was being inherited from the original pretraining configuration, so the logits shape was equivalent to the number of labels in the pretraining task

also added a .gitignore with basic settings to clean up some of the git cruft