hanlp 本地训练 gpu无法使用

hello ,我这边运行了你们的训练demo ,https://github.com/hankcs/HanLP/blob/master/plugins/hanlp_demo/hanlp_demo/zh/train_sota_bert_pku.py ,然而,在训练过程中发现,gpu未使用,cpu飙升,

hanlp

torch 可用的,cuda也是没有问题的