Files
smoltorch/examples/train_classification.py