github.com/NVIDIA/aistore@v1.3.23-0.20240517131212-7df6609be51d/docs/examples/etl-imagenet-dataset (about)

train_aistore.py
train_pytorch.py