search

avik-pal / FluxMPI.jl

Distributed Data Parallel Training of Deep Neural Networks
MIT License
56 stars 4 forks source link

scaling example for resnet50 #12

Closed dnabanita7 closed 2 years ago

dnabanita7 commented 2 years ago

refer #11

avik-pal commented 2 years ago

Thanks for the PR. I was thinking we could have a full training script using FastAI instead of just using dummy inputs.

dnabanita7 commented 2 years ago

Yes, I think that is a good idea.