pygod icon indicating copy to clipboard operation
pygod copied to clipboard

ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 64])

Open nbijlani opened this issue 3 years ago • 1 comments

When running model.fit (on GPU), I received the following error: ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 64])

Any suggestion to get rid of this problem when using on GPU is appreciated.

nbijlani avatar Oct 12 '22 14:10 nbijlani

Thanks for your issue. Can you provide us more details about the error? Which dataset are you using?

kayzliu avatar Oct 15 '22 01:10 kayzliu