I am trying to reload the weights into a custom Keras layer that I have trained in a previous run. It seems that after calling the load_weights method, not all parts of the