How to set training=False for keras-model/layer outside of the __call__ method?...