Skip to content

GitLab

Unverified Commit e46983f6 authored by Tianqi Zhang (张天启)'s avatar Tianqi Zhang (张天启) Committed by GitHub
Browse files

[Bug Fix][Example] fix dataloader bug in gxn (#2631) 


Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
parent 22ccf436
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/gxn/main_early_stop.py
View file @ e46983f6
...@@ -5,12 +5,12 @@ from datetime import datetime ...@@ -5,12 +5,12 @@ from datetime import datetime
import dgl import dgl
from dgl.data import LegacyTUDataset from dgl.data import LegacyTUDataset
from dgl.dataloading import GraphDataLoader
from torch.utils.data import random_split from torch.utils.data import random_split
import torch import torch
from torch import Tensor from torch import Tensor
import torch.nn.functional as F import torch.nn.functional as F
from dataloader import GraphDataLoader
from networks import GraphClassifier from networks import GraphClassifier
from utils import get_stats, parse_args from utils import get_stats, parse_args
from data_preprocess import degree_as_feature, node_label_as_feature from data_preprocess import degree_as_feature, node_label_as_feature
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment