Files
torch_geometric/scripts/visualize_graph.py