search

HSLCY / ABSA-BERT-pair

Utilizing BERT for Aspect-Based Sentiment Analysis via Constructing Auxiliary Sentence (NAACL 2019)
https://www.aclweb.org/anthology/N19-1035
MIT License
495 stars 138 forks source link

请问这个代码可以在1080Ti上运行吗 #16

Closed GW-S closed 5 years ago

GW-S commented 5 years ago

我尝试在24G显存的机器上运行，都会报错 报错为： RuntimeError: CUDA out of memory. Tried to allocate 144.00 MiB (GPU 0; 10.92 GiB total capacity; 10.43 GiB already allocated; 5.50 MiB free; 49.96 MiB cached)

GW-S commented 5 years ago

可以的，需要把batch_size的大小调低。