Я новичок в этой области и пытаюсь узнать по ссылке ниже Github. Тем не менее, я столкнулся с ошибкой во время выполнения. Несмотря на изменение размера пакета и накопления градиента до меньших значений и очистку кэша, ошибка времени выполнения сохраняется.
Может кто-нибудь поделиться какой-либо информацией об этом? Большое спасибо.
https://github.com/nlptown/nlp-notebooks/blob/master/Text%20classification%20with%20BERT%20in%20PyTorch.ipynb
RuntimeError: CUDA не хватает памяти. Попытка выделить 1024,00 КБ (GPU 0; общая емкость 4,00 ГБ; 3,09 ГБ уже выделено; 528,00 КБ бесплатно; 32,48 МБ кэшировано)