pygod
pygod copied to clipboard
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 64])
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.
Thanks for your issue. Can you provide us more details about the error? Which dataset are you using?