jiangnanboy / albert_lstm_crf_ner

albert + lstm + crf实体识别,pytorch实现。识别的主要实体是人名、地名、机构名和时间。albert + lstm + crf (named entity recognition)
132 stars 31 forks source link

The GPU using rate is very low #4

Open LIZ-dandan opened 2 years ago

LIZ-dandan commented 2 years ago

I ran the main.py and found that though the gpu is available, the training speed was very slow. I checked the nvidia-smi, and the using rate was only lower than 10% most of the time. How can I solve this problem?