Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Benchmark checkpoint #45

Merged
merged 11 commits into from
Jul 10, 2024
Merged

Benchmark checkpoint #45

merged 11 commits into from
Jul 10, 2024

Conversation

divrawal
Copy link
Collaborator

@divrawal divrawal commented Jul 9, 2024

Benchmarking lightning checkpoint directory that intakes environment variables to run the tests. Steps and size are configurable along with being able to toggle between TorchCheckpointIO and DatafluxLightningCheckpoint

  • Tests pass
  • Appropriate changes to documentation are included in the PR

@divrawal divrawal requested a review from a team as a code owner July 9, 2024 18:44
@divrawal divrawal requested review from MattIrv and bernardhan33 July 9, 2024 18:44
dataflux_pytorch/benchmark/README.md Show resolved Hide resolved
dataflux_pytorch/benchmark/README.md Show resolved Hide resolved
dataflux_pytorch/benchmark/README.md Outdated Show resolved Hide resolved
dataflux_pytorch/benchmark/README.md Show resolved Hide resolved
dataflux_pytorch/benchmark/README.md Outdated Show resolved Hide resolved
@divrawal divrawal merged commit 2da1582 into main Jul 10, 2024
2 checks passed
@divrawal divrawal deleted the benchmark-checkpoint branch July 10, 2024 23:10
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants