hydra-torch icon indicating copy to clipboard operation
hydra-torch copied to clipboard

Single-node ImageNet DDP

Open briankosw opened this issue 4 years ago • 4 comments

An example of DDP for ImageNet using multirun, as discussed with @omry in the Hydra repo.

briankosw avatar Nov 18 '20 02:11 briankosw

@omry, let me know if you think another issue for torch.distributed + multirun will be useful here too.

briankosw avatar Nov 18 '20 02:11 briankosw

@briankosw, since you already have the other example read you can just send a PR directly.

omry avatar Nov 18 '20 08:11 omry

@briankosw If you have a draft PR, I'm happy to check it out! =] Doesn't need to be finished.

romesco avatar Nov 24 '20 19:11 romesco

@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).

briankosw avatar Dec 01 '20 04:12 briankosw