What versions of TF and Keras are you using? Could you please provide a configured Docker container? Thank you in advance. Best.