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 from
Trackable), 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!