cora，citeseer，PubMed都有这个情况

[unu165]

Traceback (most recent call last):
File "train.py", line 176, in
main()
File "train.py", line 172, in main
train_eval(args)
File "train.py", line 166, in train_eval
trainer.train()
File "train.py", line 44, in train
dev_best, dev_std_best, test_best, test_std_best = self.evaluate()
File "train.py", line 139, in evaluate
logits, loss = classifier(self._embeddings[self._train_mask], self._labels[self._train_mask])
File "/home/ps/anaconda3/envs/bgrl/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/ssd/export/jiajie/BGRL_Pytorch-main/models.py", line 125, in forward
loss = self.cross_entropy(logits, y)
File "/home/ps/anaconda3/envs/bgrl/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/home/ps/anaconda3/envs/bgrl/lib/python3.8/site-packages/torch/nn/modules/loss.py", line 1150, in forward
return F.cross_entropy(input, target, weight=self.weight,
File "/home/ps/anaconda3/envs/bgrl/lib/python3.8/site-packages/torch/nn/functional.py", line 2846, in cross_entropy
return torch._C._nn.cross_entropy_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index, label_smoothing)
RuntimeError: Expected target size [1, 6], got [1, 3327]