AttributeError: Can‘t get attribute ‘CrfInputFeatures‘ on ＜module ‘__main__‘

这是我在跑https://github.com/hquzhuguofeng/KBQA-BERT-CRF
这个项目，torch.load文件时出的错。
找不到CrfInputFeatures 这个自定义的类。

搜了一下，都是将CrfInputFeatures 这个自定义的类 复制到该py文件中，或者import CrfInputFeatures 这个自定义的类。

然并卵。报一样的错。

解决方法：同一目录下新建新的test.py，将之前报错的test_ner中代码复制到test.py中，运行test.py，成功。

最后，也不知道是为什么，希望能有大佬解释一下。