How to get mode summary if I use tune.run()?

[sven1977]

Thanks @mannyv for your detailed answer and the really cool hack into the after_init
This is great. @bug404 I answered your question on how to do this for torch in another thread here in the forum:

How to get model summary using Pytorch backend?

Hey @bug404 . If you have an RLlib Trainer object: print(trainer.get_policy().model) If you are using tune.run, try to add this line into ray/rllib/policy/torch_policy.py:~164 (after we assign self.model to the 0th of the multi-GPU towers): print(self.model)