Error running fine tune notebooks

agemagician / ProtTrans

ProtTrans is providing state of the art pretrained language models for proteins. ProtTrans was trained on thousands of GPUs from Summit and hundreds of Google TPUs using Transformers Models.

Academic Free License v3.0

1.13k stars 153 forks source link

Error running fine tune notebooks #78

Closed BSharmi closed 2 years ago

BSharmi commented 2 years ago

Hello!

I am trying to replicate the Localization fine tune notebook (https://github.com/agemagician/ProtTrans/blob/master/Fine-Tuning/ProtBert-BFD-FineTuning-PyTorchLightning-Localization.ipynb) and get an error AttributeError: can't set attribute in the

model = ProtBertBFDClassifier(hparams) while trying to set the hparams self.hparams = hparams

Did I miss anything here? I am just rerunning the notebook

Thank you!

BSharmi commented 2 years ago

I think I needed to install the specified PyTorch version to run the notebook