e7f5220dfa
device_ids can be None again in data_parallel ( #1187 )
2017-04-06 10:30:53 -04:00
e50a1f19b3
Use streams in scatter to overlap copy with compute
2017-03-14 22:46:07 +01:00
60736bdf99
fix corner case in kwargs for DataParallel ( #930 )
2017-03-05 14:27:52 -05:00
b1ae7f90d5
Added functionality for data parallel table ( #843 )
2017-03-05 02:35:46 +01:00
88275da5e8
CUDA documentation tweaks ( #858 )
2017-02-26 20:37:43 +01:00
b87c113cf4
CUDA documentation enhancement and docs versioning ( #848 )
...
* Add more detail to CUDA documentation
Also adds better cross-linking to the pages that discuss relevant topics.
* Adds recommendation to torch.save docs
* Make the version numbers for the docs dynamic
Might need tweaks for beta, 1.0, etc.
2017-02-26 08:33:26 -05:00
876202503f
Support multiple inputs in data parallel
2017-02-20 23:28:31 -08:00
7c44506441
allow DataParallel to have tuple inputs on a single GPU
2017-02-16 19:07:17 +01:00
d6fa3b3fd5
Deprecate nn.Container in favor of nn.Module
2017-01-16 19:07:37 -05:00
ea728e7c5e
Add DataParallel container ( #268 )
...
Adds a container version of the `data_parallel` function. This is a
drop-in replacement for the DataParallel class in the ImageNet example.
2016-11-29 16:36:01 -05:00
