Error in restoring the saved model

RLlib
1

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/trainable.py”, line 783, in restore
self.load_checkpoint(to_load)
File “/usr/local/lib/python3.8/dist-packages/ray/rllib/algorithms/algorithm.py”, line 2004, in load_checkpoint
checkpoint_data = Algorithm._checkpoint_info_to_algorithm_state(
File “/usr/local/lib/python3.8/dist-packages/ray/rllib/algorithms/algorithm.py”, 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?

Thanks,

2

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.