amaze.extensions.sb3.graph¶
Collection of functions to generate a user-friendly representation of a neural network from stable baselines 3
Functions
|
Generates a (dot) graph from the underlying pytorch elements |
Back to top