Bert-Multi-Label-Text-Classification icon indicating copy to clipboard operation
Bert-Multi-Label-Text-Classification copied to clipboard

Published 20 hours ago • verified publisher icon lonePatient

GPU MEMORY

Open 1eclipse opened this issue 5 years ago • 2 comments

hello，Are there any requirements for GPU memory？

1eclipse avatar Jul 09 '20 03:07 1eclipse

In my experience,I think it requires greater than 2G GPU memory.Because I got a "out of memory" error in my computer with 2G GPU memory.

unstory avatar Aug 20 '20 01:08 unstory

bert-base+256length+16batch ~> 8g GPU memory

lonePatient avatar Aug 20 '20 06:08 lonePatient