Error in Roundtrip Tutorial: "Checkpoint was expecting d_optimizer to be a trackable object”

[rnayebi21]

Description:

I encountered an error while running the Roundtrip tutorial from the repository (cloned directly from GitHub). The error suggests that d_optimizer is not trackable when used in a TensorFlow Checkpoint. It seems to be related to Keras optimizers.

Steps to reproduce:

Running the file Roundtrip/src/tutorial.ipynb

Specifically the cell that produces the error is the third one at line:

model = pyrt.Roundtrip(params=params,random_seed=123)

Error:

ValueError: Checkpointwas expecting d_optimizer to be a trackable object (an object derived fromTrackable), got <keras.src.optimizers.adam.Adam object at 0x7f8abedf72c0>. If you believe this object should be trackable (i.e. it is part of the TensorFlow Python API and manages state), please open an issue.

Any guidance on this issue would be greatly appreciated!