Ok, thank you, i made the alteration that you suggest. I´m stuck here now. I get this errors:
File “C:/Users/user/.spyder-py3/original_cnn.py”, line 112, in 
loss = criterion(label,torch.argmax(output,dim=1))
File “C:\Users\user\Anaconda3\lib\site-packages\torch\nn\modules\module.py”, line 541, in call
result = self.forward(*input, **kwargs)
File “C:\Users\user\Anaconda3\lib\site-packages\torch\nn\modules\loss.py”, line 431, in forward
return F.mse_loss(input, target, reduction=self.reduction)
File “C:\Users\user\Anaconda3\lib\site-packages\torch\nn\functional.py”, line 2204, in mse_loss
ret = torch._C._nn.mse_loss(expanded_input, expanded_target, _Reduction.get_enum(reduction))
RuntimeError: Expected object of scalar type Float but got scalar type Long for argument #2 ‘target’ in call to _thnn_mse_loss_forward