Error locating to dgl.node_subgraph with Heterogeneous graph.
Specific code:dgl.node_subgraph(g, {‘mention’: xxx, ‘entity’: xxx})
Error description:what(): CUDA out of memory. Tried to allocate 1109057167.77 GiB (GPU 0; 23.70 GiB total capacity; 55.85 MiB already allocated; 21.73 GiB free; 70.00 MiB reserved in total by PyTorch)
I don’t know why generating subgraphs causes so many memory overflows.