I’m testing the program with your model, and I modified differents parameters, as Batch-size, width, and epoch. When I run the code, on CPU = 500sec nearly, against 70sec on GPU, I think it’s the beginnning of the solution !!  If someone has differents advices, I’m ok to test these !