I have not seen anything in the rllib documentation that would allow me to print a quick summary of the model like print(model.summary()) in keras. I tried using tf-slim and variables = tf.compat.v1.model_variables() slim.model_analyzer.analyze_vars(variables, print_info=True) to get a rough idea for tensorflow models, but this found no variables after the model was initialized (inserted at the end of the ESTrainer class _init). Specifically, I have been trying to get a summary of an Evolutionary Strategy (ES) policy to verify that the changes to the model config are being updated as expected, but I have not been able to get a summary print working. Is there an existing method for this? Is slim expected to work here? Continue reading...
