256 Commits

Author	SHA1	Message	Date
Adam Paszke	a1fa995044	Fixes and improvements (#593) ... * Fix error in ELU backward * Add --seed flag for testst st * Add test for BatchNorm eval * Fix autograd.backward docs * Support cc flags in cuDNN search * Fix IndexSelect backward formula	2017-01-25 22:21:49 -05:00
Sam Gross	ac8a5e7f0d	Remove error message assertion (#480) ... Depending on how PyTorch is compiled, the source code for DataLoader might not be fully available which can cause a spurious error in test_dataloader.py	2017-01-18 13:16:38 -05:00
Sergey Zagoruyko	89d930335b	fix tests for GPU-less setup (#298)	2016-12-12 10:56:57 +01:00
Sam Gross	be3276fcdd	Account for batch_size in DataLoader.__len__() (#277)	2016-12-02 01:21:36 -05:00
Sam Gross	aea6ba4bcd	Support pinned memory in the DataLoader (#265) ... DataLoader now supports the constructor argument 'pin_memory'. When set to true, tensors in the sample are copied to pinned memory. This happens in a background thread when num_workers > 1.	2016-11-29 12:35:03 -05:00
Sam Gross	6db721b5dd	Make DataLoader preserve the ordering of the dataset (#135)	2016-10-21 23:54:16 -04:00