pangyunsheng8 / cgipool Goto Github PK

Traceback (most recent call last):
File "/home/sc5/lizhipeng/GNN/pool/constrastive learning/CGIPool-main/train.py", line 127, in
main()
File "/home/sc5/lizhipeng/GNN/pool/constrastive learning/CGIPool-main/train.py", line 116, in main
train(args, model, train_loader, val_loader)
File "/home/sc5/lizhipeng/GNN/pool/constrastive learning/CGIPool-main/train.py", line 40, in train
output = model(data)
File "/home/sc5/anaconda3/envs/cl/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/sc5/lizhipeng/GNN/pool/constrastive learning/CGIPool-main/models/SAGNet.py", line 42, in forward
x, edge_index, _, batch, _, loss = self.pool1(x, edge_index, None, batch)
File "/home/sc5/anaconda3/envs/cl/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/sc5/lizhipeng/GNN/pool/constrastive learning/CGIPool-main/poolings/CGIPool.py", line 70, in forward
real_loss = self.loss_fn(self.discriminator(positive_pair), real)
File "/home/sc5/anaconda3/envs/cl/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/sc5/anaconda3/envs/cl/lib/python3.6/site-packages/torch/nn/modules/loss.py", line 530, in forward
return F.binary_cross_entropy(input, target, weight=self.weight, reduction=self.reduction)
File "/home/sc5/anaconda3/envs/cl/lib/python3.6/site-packages/torch/nn/functional.py", line 2519, in binary_cross_entropy
"Please ensure they have the same size.".format(target.size(), input.size()))
ValueError: Using a target size (torch.Size([128])) that is different to the input size (torch.Size([128, 1])) is deprecated. Please ensure they have the same size.

i have solved it. The correct code which should be:
real = torch.ones(positive_pair.shape[0],1).cuda()
fake = torch.zeros(negative_pair.shape[0],1).cuda()