Pytorch CUDA out of memory. Обучение RNN сети

Я написал по уроку RNN-модель (код ниже) , но при увеличении данных для обучения столкнулся с ошибкой:

Как мне решить эту проблему и продолжить обучение? Я пробовал уменьшить batch_size , писать torch.cuda.empty_cache(), уменьшал количество hidden слоев - ничего не помогает. Может можно как-то вручную дать больше памяти? Спасибо за любой ответ.

Видеокарта:

• NVIDIA GeForce GTX 1660

Код (+ text.txt - данные на которых НН обучается; text_2.txt - возникает ошибка):

• https://drive.google.com/drive/folders/1BpGowyXouf-PFn1XLsC3lJj_TjFreRiB?usp=sharing