exploring-T5 icon indicating copy to clipboard operation
exploring-T5 copied to clipboard

t5 training notebook issue

Open riyaj8888 opened this issue 4 years ago • 3 comments

Hello Suraj , Thanks for sharing this fine tuning of t5 notebook. I am trying to use it as it is but when i am running it on colab it throwing following error.

**trainer = pl.Trainer(train_params) "TypeError: init() got an unexpected keyword argument 'early_stop_callback'"

I will appreciate your time in resolving this issue.

thanks

riyaj8888 avatar Jan 15 '21 05:01 riyaj8888

Hi! I had the same problem and I figured out that it was a package version problem. In order to make this notebook work properly, you need to use this versions:

!pip install transformers==2.9.0 
!pip install pytorch_lightning==0.7.5

MarcosFP97 avatar Jan 27 '21 10:01 MarcosFP97

I have created a PR, but meanwhile you can download the fixed notebook from my fork: here

Best, Marcos

MarcosFP97 avatar Jan 27 '21 10:01 MarcosFP97

Hi, you can simply comment the early_stop_callback and it will work

Jackthebighead avatar Nov 09 '21 07:11 Jackthebighead