Values error when using BioSNAP dataset

yueyu1030 / SumGNN

[Bioinformatics 2021] This is the repo for the paper `SumGNN: Multi-typed Drug Interaction Prediction via Efficient Knowledge Graph Summarization'.

https://academic.oup.com/bioinformatics/advance-article/doi/10.1093/bioinformatics/btab207/6189090

81 stars 20 forks source link

Values error when using BioSNAP dataset #8

Open Layne-Huang opened 2 years ago

Layne-Huang commented 2 years ago

When I want to run the model on BioSNAP dataset, it threw out the value error: Using a target size (torch.Size([178, 178])) that is different to the input size (torch.Size([178, batch_size])) is deprecated. Please ensure they have the same size, where 178 is the number of samples in last batch.