Some datasets have not come with node labels #6
Description
Some datasets with both node lables and graph lables can run the code successfully, but there are some datasets which have not come with node labels, when I run the code, exist a bug.
Traceback (most recent call last):
File "main.py", line 306, in
run_exp_single_test()
File "main.py", line 301, in run_exp_single_test
run_exp_lib([('REDDIT-MULTI-12K', 'data', 'ResGFN')])
File "main.py", line 189, in run_exp_lib
logger=logger)
File "/home/lvshengze/gfn-master/train_eval.py", line 110, in cross_validation_with_val_set
model, optimizer, train_loader, device)
File "/home/lvshengze/gfn-master/train_eval.py", line 198, in train
out = model(data)
File "/opt/anaconda3/envs/lsz/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, **kwargs)
File "/home/lvshengze/gfn-master/res_gcn.py", line 117, in forward
return self.forward_BNConvReLU(x, edge_index, batch, xg)
File "/home/lvshengze/gfn-master/res_gcn.py", line 140, in forward_BNConvReLU
x = self.bn_feat(x)
File "/opt/anaconda3/envs/lsz/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, **kwargs)
File "/opt/anaconda3/envs/lsz/lib/python3.6/site-packages/torch/nn/modules/batchnorm.py", line 85, in forward
self._check_input_dim(input)
File "/opt/anaconda3/envs/lsz/lib/python3.6/site-packages/torch/nn/modules/batchnorm.py", line 177, in _check_input_dim
if input.dim() != 2 and input.dim() != 3:
AttributeError: 'NoneType' object has no attribute 'dim'
I have no idea to use torch_geometric.transforms.Constant or torch_geometric.transforms.OneHotDegree to add node labels to the datasets.
Sincerely looking forward to your reply.