### Train Module

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