BERT-BiLSTM-CRF-NER icon indicating copy to clipboard operation
BERT-BiLSTM-CRF-NER copied to clipboard

显存几乎沾满,利用率却很低

Open zhishui3 opened this issue 2 years ago • 2 comments

不管数据量有多大,显存本身24000MB 多 占用了23000MB多(并且这个数量不变),请问这是怎么回事啊? 用了各种方式都不管用:1、tf.data.TFRecordDataset.cache(),2、tf.data.TFRecordDataset.shard,3、把tf_record 分成多份读取,4、epoch 和batch_size 分别改成 1和16, 最后只能用 session_config.gpu_options.per_process_gpu_memory_fraction @macanv 求大佬指点!

zhishui3 avatar Aug 16 '21 07:08 zhishui3

俺也一样

sharipp132 avatar Jul 14 '23 12:07 sharipp132

请问你验证用了多久?

Anthony-BGST avatar Sep 12 '23 07:09 Anthony-BGST