Hi!
I was trying to reproduce the results on the proteins
dataset on a V100 GPU, and I am running into a few problems. First I had some issues with the BatchSampler (samplers cannot be passed to iterative datasets), so I just removed it (I am using the latest version of dgl, since 0.6 is not available with my CUDA version).
After fixing that, I came across this error message
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 64])
in line 98 of models.py
https://github.com/tech-srl/how_attentive_are_gats/blob/b5ccb6129878bb6f6f8c3f77934046d8e0920a52/proteins/models.py#L90-L100
Is this error related with removing the BatchSampler? And which tensor size is it expected at that line?
Thanks for the help