Oh Thanks!
I will try it.
I have two more question.

batch size = 2 is harder to training than batch size = 1?
My training loss = 0.002 but If I test the model with training data. loss increases to 0.2.

I suspect that dropout does not work in model.eval (), so the output value of the model has changed, or the number of epochs is insufficient.
Please help me!
Thank you.