Error when saving model with PolicySaver

[enriquegit]

The error happens when trying to save a model with tensorflow-agents. For example:

collect_policy = tf_agent.collect_policy
saver = PolicySaver(collect_policy, batch_size=None)

and saving the model with:

saver.save('models/ddpg/')

This works ok in google colab but the following error (see image below) is shown when running on a local windows PC.

library versions: tensorflow 2.9.1 tf-agents 0.11.0

[sguada]

Can you try updating using pip install tf_agents[reverb]?

[enriquegit]

@sguada I tried your suggested solution. Now, the following error is shown:

ImportError: cannot import name 'network' from 'tensorflow.python.keras.engine'

[sguada]

Probably due to mismatch version of TF.