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
3cbe66ba8c
Change requires_grad default to False
2016-10-05 08:46:34 -07:00
Adam Paszke
1d0afdf9f7
Make requires_grad read only (except for leaves)
2016-10-05 07:55:07 -07:00
Adam Paszke
64dd1419c5
Fix Variable indexing bugs ( #96 )
2016-10-03 14:49:21 -04:00
Adam Paszke
11b38a6895
Add more functions to autograd
2016-09-30 16:37:07 -04:00
Adam Paszke
f9d9c92560
Fix type conversions in autograd
2016-09-27 15:45:52 -07:00
Adam Paszke
1828e7c42f
Add async CUDA copy
2016-09-27 15:12:48 -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
8fdec15a55
Codemod to remove camel case method naming
2016-09-20 08:40:28 -07:00
Adam Paszke
7847d77405
Add more functions to autograd
2016-09-16 15:26:24 -07:00
Adam Paszke
4bad029fd4
Add more functions to autograd
2016-09-15 13:01: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
24476090df
Add volatile variables
2016-08-24 08:43:11 -07:00
Adam Paszke
ea93fb7ac0
Add more nn modules
2016-08-23 19:15:21 -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