- Document the training crops used for the model. As a csv table that's saved alongside the checkpoint? - Add argument for save_root, to modify where the model is saved. - Suppress warnings.
