Add support for multithreaded training in the neural net example#2454
Add support for multithreaded training in the neural net example#2454WilliamTambellini wants to merge 5 commits intoarrayfire:masterfrom
Conversation
Enhance the neural net example with multigpu training in parallel.
|
The updated example looks good. I have made some cmake changes. |
|
Cool tks. Feel free to merge/modify your way. |
|
I ran it on multiple(two: GTX 1060, AMD Spectre R7) devices, works without any issues. This change demonstrates how to use multiple devices but not so much on how to distribute training data across multiple devices. How difficult would be such an example to implement ? |
|
|
That item would allow to also compare speed of mutiple vs single gpus execution. |
Add support for multigpu parallel training in the neural net example