### Utils Module

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