github.com/kubeflow/training-operator@v1.7.0/examples/pytorch/elastic/imagenet (about) .dockerignore Dockerfile imagenet.py imagenet.yaml