Download and prepare the data To run the code Within the container, run source config_DGXA100_1x8x56x1.sh bash run_with_docker.sh