search

thunlp / EntityDuetNeuralRanking

Entity-Duet Neural Ranking Model
MIT License
153 stars 20 forks source link

CUDA out of memory #17

Closed gaoliming123 closed 3 years ago

gaoliming123 commented 3 years ago

您好：我用pytorch1.4版本跑您的代码发现跑到几千个step之后就会出现CUDA out of memory image 方便说一下您代码的环境配置吗？