Yes, I set model.train() and model.eval().
I also checked the outpus, they are the same on the same input.
I also execute one step of the optimization method.
After update, the first conv layer kernel keep the same.
I did’t execute more steps.