@ptrblck @gphilip @BruceDai003  Thanks for you reply.
I followed your suggestion, but unfortunately it is still not training…I checked loss and weights that have different value in each epoch, but the model is not learning within 50 epoches. Besides, I also tried different model, architecture and hyperparameters. The code is as following link, can you please help to check the code? Appreciate your help.

  
      

      GitHub
  

  
    

GitHub - yellow07200/Test 4

  Contribute to yellow07200/Test development by creating an account on GitHub.