Skip to content
Build Image

Support writing optimizer checkpoint only on rank0 and make UT pass on A100 #355

Sign in to view logs

Support writing optimizer checkpoint only on rank0 and make UT pass on A100

Support writing optimizer checkpoint only on rank0 and make UT pass on A100 #355

Run time

Learn about OS pricing on GitHub Actions
Job Run time
Docker build torch1.14-cuda11.8
18s
Docker build torch2.1-cuda12.2
1s
19s