Skip to content

Update for latest Horovod and add new GRACE Horovod example#25

Open
Quentin-Anthony wants to merge 2 commits intosands-lab:masterfrom
Quentin-Anthony:latest-horovod
Open

Update for latest Horovod and add new GRACE Horovod example#25
Quentin-Anthony wants to merge 2 commits intosands-lab:masterfrom
Quentin-Anthony:latest-horovod

Conversation

@Quentin-Anthony
Copy link
Copy Markdown

@Quentin-Anthony Quentin-Anthony commented Sep 8, 2022

This PR updates the current patch files to conform to Horovod's new horovod/torch/optimizer.py structure. Tested and verified with Horovod 0.23.0 and PyTorch 1.7.1 on the new imagenet example (examples/torch/pytorch_imagenet_resnet50-grace.py).

I'm aware that the GRACE team is working on a cleaner way to integrate with Horovod (#21), this is just to maintain support in the meantime.

Let me know if you have any questions or edits :)

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.

1 participant