HeterSumGraph
HeterSumGraph copied to clipboard
dqwang122
Code for ACL2020 paper "Heterogeneous Graph Neural Networks for Extractive Document Summarization"
CODE: !python HeterSumGraph/evaluation.py --cuda --gpu 1 --data_dir /content/dataset/cnndm --cache_dir /content/graphfile/cache/CNNDM --embedding_path /content/glove.6B.300d.txt --model HSG --save_root /content/models/ --log_root /content/logs -m 3 --test_model multi --use_pyrouge ERROR: File "/content/HeterSumGraph/evaluation.py", line 239, in main()...
My ROUGE installation should be fine as I have no problem with the CNN/DailyMail dataset at all, but the ROUGE score on the Multi-News dataset is: Rouge1 =40.4, RougE2 =15.7,...
 Traceback (most recent call last): File "E:\project\HeterSumGraph-master\train.py", line 438, in main() File "E:\project\HeterSumGraph-master\train.py", line 385, in main vocab = Vocab(VOCAL_FILE, args.vocab_size) File "E:\project\HeterSumGraph-master\module\vocabulary.py", line 53, in __init__...
Just wanted to know does the changes in GAT layer in the dev branch actually improve the performance of the model (ROUGE scores) ?
运行过程中，好像需要将G.to_device("cuda")改为G=G.to_device("cuda")，否则会报错RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking argument for argument index in method wrapper__index_select)。不知道是不是我自己的原因。
Dear author, can I ask you a detail on the implementation? Did the training stop early when you got the expected performance of the model? If yes, and what is...
