### Models Module

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