Dhaval Taunk

icon indicating a website link https://dhavaltaunk08.github.io/

icon company Fast Code AI icon location Bengaluru, India icon location A data science practitioner, computer science geek having an aim to solve real-world problems

Results 4 comments of Dhaval Taunk

Multi-gpu support

Can you please tell me which model you are referring to DistilBERT or Roberta?

Multi-gpu support

Please add the code and error in the comments. It is difficult to check errors from the image.

How to load the fine-tuned model after save it local.

You are missing your config.json file. Make sure you have the files placed at the correct location.

GPU memory

Hi @MarkusSagen, I am using my own custom built dataset having same format as Wikisum dataset. I tried different batch sizes varying from 2 to 16. I have tried sequence...