katyov

ValueError: Target size (torch.Size([4, 2])) must be the same as input size (torch.Size([8, 2])

3

SageMaker training with imdb sample data throws the error: ValueError: Target size (torch.Size([4, 2])) must be the same as input size (torch.Size([8, 2]) hyperparameters = { "epochs": 6, "lr": 5e-3,...