how to input text and display the results by DiffusionNER Model?

[StanleySun233]

After the model training is completed, how to input text and display the results? Where do I need to modify the code to implement this functionality?

[xiaoli]

Let's examine lines 23-26 in diffusionner.py. Then, we can delve into the diffusionner_trainer.py file, specifically lines 214 and beyond, and evaluator.py starting at line 159. The variable pred_entities contains the predicted results, which I believe is what you're looking for.

[houyuchao]

Let's examine lines 23-26 in diffusionner.py. Then, we can delve into the diffusionner_trainer.py file, specifically lines 214 and beyond, and evaluator.py starting at line 159. The variable pred_entities contains the predicted results, which I believe is what you're looking for.

您遇到过这样的问题吗，这是为什么

[xiaoli]

@houyuchao 这个问题，我没遇到过。

看提示，应该是所有的GPU都正在被使用，所以它们都非常忙，你需要一个有空的GPU来执行你的程序代码。

[modenicheng]

Let's examine lines 23-26 in diffusionner.py. Then, we can delve into the diffusionner_trainer.py file, specifically lines 214 and beyond, and evaluator.py starting at line 159. The variable pred_entities contains the predicted results, which I believe is what you're looking for.

您遇到过这样的问题吗，这是为什么

@houyuchao 如果你 GPU 确认是没有占用的，那么就是你的总显存不够。参见 config_reader.py 第 51-54 行，这里定义了所需要的显存大小。如果你非要在低显存的情况下跑，把 11000 改小就行了