Keras does not save weights properly

[gcelano]

I am using Tensorflow 2.15.1 and, if I use model.save_weights("./dir/mymodel.bin") and subsequently try to load the weights with model.load_weights (in a different process), the model does not return the expected results (when using model.evaluate or .predict). My model has a custom loss function and metric, but this should not be a problem, since I try to load weights only. Any idea why I get this error?

[sachinprasadhs]

Could you please try with the Keras 3 version and let us know your observations and possibly some reproducible code. Thanks

[gcelano]

I installed Tensorflow 2.16.1, but the it cannot detect the GPUs.

[gcelano]

The problem is also with Tensorflow 2.16. Please, run the code here:

https://git.informatik.uni-leipzig.de/celano/dithrax_test