Adam Lerer
b5d13296c6
addressing comments
2016-10-23 21:11:22 -07:00
Adam Lerer
942ca477a6
Copying weights for CUDNN
2016-10-23 20:23:48 -07:00
Adam Paszke
0325e2f646
Major autograd refactor
...
Improves autograd performance by more than 2x and fixes a couple
of bugs. All core functions have been moved to C.
2016-10-13 17:17:49 -07:00
Adam Paszke
a22af69335
Add versioning and shared storage handling to autograd ( #105 )
2016-10-06 17:12:58 -04:00
Adam Paszke
1d0afdf9f7
Make requires_grad read only (except for leaves)
2016-10-05 07:55:07 -07:00
Sam Gross
93e60715af
Fix error message
2016-09-29 16:27:20 -07:00
Adam Paszke
4742c08c7c
Improve error messages in autograd
2016-09-29 12:16:19 -07:00
Sam Gross
980300b381
Combine autograd.Leaf and autograd.Variable ( #52 )
...
Prior to this change, there was a circular reference between Leaf and
Variable. This means that the objects (and referenced Tensors) are not
collected as soon as they go out of scope, which lead to higher memory
usage and out-of-memory errors.
2016-09-25 20:21:14 -04:00
Adam Paszke
7847d77405
Add more functions to autograd
2016-09-16 15:26:24 -07:00
Adam Paszke
fb39971464
Add more modules to nn
2016-09-14 11:05:56 -07:00
Sam Gross
b738b09606
Clean up Module forward and __call__ ( #14 )
...
* _forward is renamed forward since users should override it
* some __call__ overrides are changed to forward
* function which return a single variable are changed to return that
variable instead of a one-element tuple
2016-09-07 15:41:39 -04:00
Adam Paszke
774a6f1093
Add in-place operations to autograd and nn
2016-08-25 09:34:54 -07:00
Adam Paszke
814728a6aa
Improve hook system
2016-08-25 09:23:39 -07:00
Adam Paszke
24476090df
Add volatile variables
2016-08-24 08:43:11 -07:00
Adam Paszke
2bf68e72d5
Add hook system to autograd and nn
2016-08-23 13:51:34 -07:00
Adam Paszke
53f00ae429
Add autograd
2016-08-19 14:24:07 -07:00