Error saving model about deepsphere-cosmo-tf2 HOT 6 CLOSED

Dear Javier

Thank you very much for your interest in DeepSphere for TF 2.x

I was able to reproduce the error regarding model.save, I think that there was an issue with the subclassing. However, I did not manage to reproduce the same error with model.save_weights .

Anyway, I pushed a potential fix, can you check if a8211be fixed your issue?

Best,

Janis

from deepsphere-cosmo-tf2.

It works now!! I have tested this new version with model.save, model.save_weights along with model.load &model.load_weights, and everything seems to be working. Thank you very much @jafluri for your support.
Best,
Javier

from deepsphere-cosmo-tf2.

Hello Janis,
Thank you for your quick reply. Unfortunately, I'm still having a similar error:

10/10 [==============================] - ETA: 0s - loss: 1.1198 - sparse_categorical_accuracy: 0.6933
Epoch 1: val_loss improved from inf to 0.69315, saving model to my_model/cp-0001.ckpt
/usr/local/lib/python3.7/dist-packages/keras/utils/traceback_utils.py in error_handler(*args, **kwargs)
65 except Exception as e: # pylint: disable=broad-except
66 filtered_tb = _process_traceback_frames(e.traceback)
---> 67 raise e.with_traceback(filtered_tb) from None
68 finally:
69 del filtered_tb

/usr/lib/python3.7/contextlib.py in exit(self, type, value, traceback)
117 if type is None:
118 try:
--> 119 next(self.gen)
120 except StopIteration:
121 return False

TypeError: chebyshev_layer_call_and_return_conditional_losses(input_tensor, training) got two values for 'training'.

In the case of using model.save_weights, I've got the error when loading: model.load_weights(latest). I am running in a local conda env. and Google-colab, but in both environments I got the same error.

from deepsphere-cosmo-tf2.

I see, thanks for the report.

Which version of TensorFlow are you running?

from deepsphere-cosmo-tf2.

Anyway, I tried something new in 555e28e, let me know if it worked.

from deepsphere-cosmo-tf2.

I see, thanks for the report.

Which version of TensorFlow are you running?

TF. 2.8.0

from deepsphere-cosmo-tf2.