Thanks a lot. It works now but I think the training process is too slow in compare with before I did’t have validation and scheduling the learning rate.  4 epoch and step_size=2 completed in 7m. I will post a new thread for that as I have some questions about the scheduling the learning rate.