How to improve training performances of image Colorization model using Auto-encoders

Question

I\'m working on an Image colorization model using auto-encoders,Here are parameters I\'m using :

• Training images size:26000

• Validation image