关于 data cache 生成的问题
Youggls opened this issue · 0 comments
Youggls commented
尊敬的作者您好
关于每个数据集中生成的缓存文件 kb_cache.json
m2n_cache.json
query_cache.json
,我有些小小的疑问
在您的代码文件 code/KBQA_Runner.py
中的 701-705 行,有这样几行被注释掉的代码:
model_to_save = policy.module if hasattr(policy, 'module') else policy
torch.save(model_to_save.state_dict(), save_model_file)
Save_KB_Files(convert_json_to_save(KB), save_kb_cache)
Save_KB_Files(M2N, save_m2n_cache)
Save_KB_Files(list(QUERY), save_query_cache)
在我看来,这是在训练过程中动态生成以上三个缓存文件,然后将其保存。
但是当我在使用我本人的数据集(非freebase)时,训练过程中生成的这三个文件确是空的,是否需要用其他方式手动初始化?
十分期待您的回复,非常感谢!