HeterSumGraph
HeterSumGraph copied to clipboard
Cuda version is not compatible with GPU while running the code in T4 GPU runtime of colab.
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