需要多大的显存 1080ti 爆显存了

z814081807 / DeepNER

天池中药说明书实体识别挑战冠军方案；中文命名实体识别；NER; BERT-CRF & BERT-SPAN & BERT-MRC；Pytorch

920 stars 229 forks source link

需要多大的显存 1080ti 爆显存了 #12

Closed SomeoneNotLikeYou closed 3 years ago

SomeoneNotLikeYou commented 3 years ago

RuntimeError: CUDA out of memory. Tried to allocate 384.00 MiB (GPU 0; 11.00 GiB total capacity; 9.52 GiB already allocated; 124.50 MiB free; 9.54 GiB reserved in total by PyTorch)

z814081807 commented 3 years ago

RuntimeError: CUDA out of memory. Tried to allocate 384.00 MiB (GPU 0; 11.00 GiB total capacity; 9.52 GiB already allocated; 124.50 MiB free; 9.54 GiB reserved in total by PyTorch)

尝试调小batch_size