@ptrblck Hi, please I just found out I was running this command befor the training 
torch.set_grad_enabled(False)
torch.backends.cudnn.enabled = True
can it be the reason why my model is not learning ?
if so should I delet it or modify it?
Thanks in advance