Error in restoring the saved model

Hello every one,

I have trained an agent and I saved the model. Now, I move the saved model on another system, I get this error:

File “/usr/local/lib/python3.8/dist-packages/ray/tune/trainable/”, line 783, in restore
File “/usr/local/lib/python3.8/dist-packages/ray/rllib/algorithms/”, line 2004, in load_checkpoint
checkpoint_data = Algorithm._checkpoint_info_to_algorithm_state(
File “/usr/local/lib/python3.8/dist-packages/ray/rllib/algorithms/”, line 2562, in _checkpoint_info_to_algorithm_state
worker_state[“policy_states”][pid] = pickle.load(f)
ModuleNotFoundError: No module named ‘tensorflow.python.trackable’

Could you help me how to solve this problem?


You might want to reinstall tensorflow.
Make sure that Tensorflow is up to date with Rllib’s requirements.
Please also make sure that the checkpoint is from the same Ray version.