hydra-torch
hydra-torch copied to clipboard
Single-node ImageNet DDP
An example of DDP for ImageNet using multirun, as discussed with @omry in the Hydra repo.
@omry, let me know if you think another issue for torch.distributed
+ multirun will be useful here too.
@briankosw, since you already have the other example read you can just send a PR directly.
@briankosw If you have a draft PR, I'm happy to check it out! =] Doesn't need to be finished.
@omry @romesco sorry for the delay, I've transcribed the ImageNet DDP code and added Hydra configurations to it (a lot of it was inspired by the helpful MNIST example).