Não sei como reveter este erro para realizar o exercicio ja que a função foi modificada na versão do python 3.7. criterion = nn.MSELoss().to(device) loss = criterion(pred.squeeze(), Ytns)
#/usr/local/lib/python3.7/dist-packages/ipykernel_launcher.py:13: UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument.