Richard Zou
cac3026b35
Fix typo in DataParallel docs ( #5268 )
2018-02-15 23:02:26 +01:00
Tongzhou Wang
805639906a
Broacast output requires_grad if only corresponding input requires_grad ( #5061 )
2018-02-05 23:38:35 -05:00
Nintorac
2e42272cc1
Make DataParallel a no-op when CUDA not available ( #3318 )
2017-10-29 13:47:36 +01:00
SsnL
de1f4e69dd
raw text ( #3327 )
2017-10-28 01:24:02 +05:30
Adam Paszke
421607a935
DataParallel device_ids slicing fixes ( #2200 )
2017-07-26 01:54:38 +05:30
Adam Paszke
dc17fb68e4
Fix minor bug in parallel_apply ( #2193 )
2017-07-25 03:45:00 +05:30
Adam Paszke
4af40e3471
Let parallel_apply accept arbitrary inputs
2017-07-20 01:45:57 -04:00
Adam Paszke
12813b88f6
Add DistributedDataParallel
2017-06-12 22:00:22 -04:00
Soumith Chintala
e7f5220dfa
device_ids can be None again in data_parallel ( #1187 )
2017-04-06 10:30:53 -04:00
Sam Gross
e50a1f19b3
Use streams in scatter to overlap copy with compute
2017-03-14 22:46:07 +01:00
Soumith Chintala
60736bdf99
fix corner case in kwargs for DataParallel ( #930 )
2017-03-05 14:27:52 -05:00
Christian Sarofeen
b1ae7f90d5
Added functionality for data parallel table ( #843 )
2017-03-05 02:35:46 +01:00
Eli Stevens
88275da5e8
CUDA documentation tweaks ( #858 )
2017-02-26 20:37:43 +01:00
Eli Stevens
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
Adam Paszke
876202503f
Support multiple inputs in data parallel
2017-02-20 23:28:31 -08:00
Natalia Gimelshein
7c44506441
allow DataParallel to have tuple inputs on a single GPU
2017-02-16 19:07:17 +01:00
Adam Paszke
d6fa3b3fd5
Deprecate nn.Container in favor of nn.Module
2017-01-16 19:07:37 -05:00
Sam Gross
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