### Visualize Module

```{eval-rst}
.. automodule:: visualize_training.visualize
   :members:
```