arrival-ltd / catalyst-rl-tutorial

Using Catalyst.RL to train a robot to perform peg-in-hole insertion in simulation.
MIT License
151 stars 14 forks source link

about reward visualization #10

Open Mickeyyyang opened 2 years ago

Mickeyyyang commented 2 years ago

Which file has content about the visualization of the network structure and related parameters, there is a reward curve graph at the end of the README.md. And the size of the convolution kernel of image_net is 3*3 by default?