使用demo_kg训练问题

[swo026]

将ConvE_*.yaml改成demo_kg数据集之后数据集似乎不能正确加载，会出现警告
D:\Users\lenovo\anaconda3\envs\deepKE\lib\site-packages\pytorch_lightning\utilities\data.py:122: UserWarning: DataLoader returned 0 length. Please make sure this was your intention.
rank_zero_warn(
D:\Users\lenovo\anaconda3\envs\deepKE\lib\site-packages\pytorch_lightning\utilities\data.py:153: UserWarning: Total length of CombinedLoader across ranks is zero. Please make sure this was your intention.
rank_zero_warn(
导致模型不能训练，也没有参数保存下来，最终报错
Traceback (most recent call last):
File ".\main.py", line 112, in
main()
File ".\main.py", line 107, in main
lit_model.load_state_dict(torch.load(path)["state_dict"])
File "D:\Users\lenovo\anaconda3\envs\deepKE\lib\site-packages\torch\serialization.py", line 594, in load
with _open_file_like(f, 'rb') as opened_file:
File "D:\Users\lenovo\anaconda3\envs\deepKE\lib\site-packages\torch\serialization.py", line 230, in _open_file_like
return _open_file(name_or_buffer, mode)
File "D:\Users\lenovo\anaconda3\envs\deepKE\lib\site-packages\torch\serialization.py", line 211, in init
super(_open_file, self).init(open(name, mode))
FileNotFoundError: [Errno 2] No such file or directory: ''

[Modberge]

你好，demo_kg是一个非常小的数据集，仅包含15个实体和7个关系，如果将train_bs和eval_bs参数设置过大，会发生越界错误。据我所知，ConvE_*.yaml中设置的参数不适合训练demo_kg数据集，请尝试更改train_bs或eval_bs参数，