Add ability to train on MolNet from scratch (without pretraining)

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

Closed gabegrand closed 3 years ago

gabegrand commented 3 years ago

Changes model_dir to pretrained_model_name_or_path for clarity
If pretrained_model_name_or_path is not passed in, trains a new model from scratch
Adds flags for RobertaConfig params for non-pretrained models