notebooks
notebooks copied to clipboard
CUDA out of memory Error when run trainer.hyperparameter_search()
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?
I have the same issue
me too.