DIsable to predict properties by fairseq-generate

Trước tiên, hãy tái hiện lệnh fairseq-generate và traceback đã được báo cáo. Truy vết quá trình thiết lập qua fairseq_cli/generate.py, fairseq/tasks/graph_prediction.py và fairseq/data/pyg_datasets/pyg_dataset_lookup_table.py, tập trung vào lý do pcqm4m bị từ chối. Công việc được xem là hoàn tất khi lệnh prediction được ghi trong tài liệu không còn gây ra lỗi về tên dataset này và property prediction có thể tiếp tục.

After training on pcqm4mv2, I wanted to use checkpoint to predict properties. So I used
fairseq-generate $data_binarization --task graph_prediction --path ckpts/checkpoint_best.pt --num-classes 1
The bug was shown as:
Using backend: pytorch Traceback (most recent call last): File "/home/nv3090x2/下载/ENTER/envs/graphormer/bin/fairseq-generate", line 8, in <module> sys.exit(cli_main()) File "/home/nv3090x2/下载/ENTER/lib/python3.9/site-packages/fairseq_cli/generate.py", line 413, in cli_main main(args) File "/home/nv3090x2/下载/ENTER/lib/python3.9/site-packages/fairseq_cli/generate.py", line 50, in main return _main(cfg, sys.stdout) File "/home/nv3090x2/下载/ENTER/lib/python3.9/site-packages/fairseq_cli/generate.py", line 83, in _main task = tasks.setup_task(cfg.task) File "/home/nv3090x2/下载/ENTER/lib/python3.9/site-packages/fairseq/tasks/__init__.py", line 46, in setup_task return task.setup_task(cfg, **kwargs) File "/home/nv3090x2/下载/ENTER/lib/python3.9/site-packages/fairseq/tasks/graph_prediction.py", line 179, in setup_task return cls(cfg) File "/home/nv3090x2/下载/ENTER/lib/python3.9/site-packages/fairseq/tasks/graph_prediction.py", line 155, in __init__ self.dm = GraphormerDataset( File "/home/nv3090x2/下载/ENTER/lib/python3.9/site-packages/fairseq/data/dataset.py", line 84, in __init__ self.dataset = PYGDatasetLookupTable.GetPYGDataset(dataset_spec, seed=seed) File "/home/nv3090x2/下载/ENTER/lib/python3.9/site-packages/fairseq/data/pyg_datasets/pyg_dataset_lookup_table.py", line 102, in GetPYGDataset raise ValueError(f"Unknown dataset name {name} for pyg source.") ValueError: Unknown dataset name pcqm4m for pyg source.
I really hope you can help me with the bug. Thank you very much.