search

songhaoyu / BoB

The released codes for ACL 2021 paper 'BoB: BERT Over BERT for Training Persona-based Dialogue Models from Limited Personalized Data'
https://aclanthology.org/2021.acl-long.14/
Apache License 2.0
136 stars 24 forks source link

Question about CUDA out of memory #21

Open slptongji opened 1 year ago

slptongji commented 1 year ago

I run the code on a 10 GB GPU but it seems to have the problem of CUDA out of memory. I tried to reduce the batch_size to 1 but it still doesn't work.

yahiko-l commented 1 year ago

In fact, you should replace it with an accelerated device with more memory, such as a 3090 or 4090.