charles nainan

Results 6 comments of charles nainan

Hi there, As it stands in your snippet now m1 is still hanging onto GPU memory. So what i would try is either: 1) do a del m1 after the...

Also I don't know if you are running in a jupyter or notebook or a script, but i have noticed the memory utilization to be better and more predictable in...

If you want to continue training the model from a previous finetuned checkpoint you can use the 'load_at_start flag' in fit and set it to False. This isn't documented since...

Hi. I was not thinking of adding a multi-label classification task. I can look into it though. Is there a particular open source NLP dataset you are thinking about?

Thankyou. That sounds like a great idea that will be useful to others as well.

I haven't heard anything. Let's see what EvanMu96 thinks... There was a great medium blog post at the beginning of the year on this though: https://medium.com/huggingface/multi-label-text-classification-using-bert-the-mighty-transformer-69714fa3fb3d It shouldn't be too...