search

tricktreat / DiffusionNER

Code for the paper "DiffusionNER: Boundary Diffusion for Named Entity Recognition", accepted at ACL 2023.
95 stars 17 forks source link

训练 #21

Closed houyuchao closed 3 months ago

houyuchao commented 5 months ago

有没有遇到这个问题的，明明GPU空闲却无法训练呢？？ image

Eleven-Owl commented 3 months ago

我也是 他没用cuda我不知道为什么

tricktreat commented 3 months ago

需要调整一下这里： https://github.com/tricktreat/DiffusionNER/blob/4727ebb87a5f04c448aca6463da2d03ed4f84f02/config_reader.py#L54