notebooks icon indicating copy to clipboard operation
notebooks copied to clipboard

CUDA out of memory Error when run trainer.hyperparameter_search()

Open Xiaoping777 opened this issue 4 years ago • 2 comments
trafficstars

hi here, I got the following error when run trainer.hyperparameter_search() on databricks: RuntimeError: CUDA out of memory. Tried to allocate 300.00 MiB (GPU 0; 11.17 GiB total capacity; 10.18 GiB already allocated; 274.44 MiB free; 10.50 GiB reserved in total by PyTorch)

my dataset is very small, 60 sentences in total, training epoch =20, training batch and eval batch are 8 and 2 respectively after the trial 5, I got this message, any idea to solve this problem?

Xiaoping777 avatar Oct 26 '21 02:10 Xiaoping777

I have the same issue

eagle705 avatar Nov 11 '21 06:11 eagle705

me too.

dan9thsense avatar Jan 01 '23 01:01 dan9thsense