Closed WEIYanbin1999 closed 1 year ago
Thanks for you answer. Besides, I try to run the pre-train code: "python run_pretrain.py --pretrain_dataset BIG --dataset_name BIG --num_hidden_layers 4 --train_bs 16 --lr 1e-4 --epochs 10" but the process frozen after "The two_hop_triple_path not exists, generate and dump it":
could you help me?
When first run, it needs to generate and save subgraphs file, this process may take a few more moments to complete.
It refers to the mask token, the same as the mask token used in subgraph Pre-training.