Solved by H_MP in post #3 
              
              
                Thanks a lot but I found my problem. I should put model.train() inside the training loop, but it was outside the loop and model stuck in evaluation mode (( model.eval()))