Adam Paszke
c8a4734b97
Add RReLU to both nn packages
2016-09-29 11:33:34 -07:00
Adam Paszke
2d8c2972ae
Only allow leaf variables as module parameters
2016-09-29 11:31:26 -07:00
Sam Gross
f5a6a3b0e9
Fix torch.nn.Module._apply with None types ( #66 )
2016-09-28 19:31:07 -04:00
Sam Gross
bab7f89cdc
Fix no_bias constructor for conv2d ( #65 )
2016-09-28 19:30:43 -04:00
Soumith Chintala
412019dbe4
fixing CPU builds by making cuda imports optional
2016-09-28 11:56:18 -04:00
Adam Paszke
7f4ff0e615
Fix type conversions in nn
2016-09-27 15:45:49 -07:00
Adam Paszke
3eac7164f4
Add data parallel functions to nn
2016-09-27 15:45:45 -07:00
Sam Gross
44481354fc
Add back support for child=None in Container constructor ( #55 )
...
It's often useful to have optional child modules, such as the
downsampling operation in ResNets. Add a test for this case:
nn.Container(
child=None,
)
2016-09-26 17:18:02 -04: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
4cdeae3283
Return only unique variables from parameters()
2016-09-25 12:23:43 -07:00
Adam Paszke
8fdec15a55
Codemod to remove camel case method naming
2016-09-20 08:40:28 -07:00
Adam Paszke
d1fda539b7
Fix nn serialization errors
2016-09-15 19:28:34 -07:00
Adam Paszke
fb39971464
Add more modules to nn
2016-09-14 11:05:56 -07:00
Sam Gross
cd0929aa5e
Use chainer-style constructor for Conv2d
...
* Conv2d, MaxPool2d, and AvgPool2d have one argument for each of ksize,
stride, and pad. This argument can be either a single number or a
tuple of (h, w)
2016-09-07 15:51:44 -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
d467a068c2
Add tests for new modules
2016-08-19 14:57:01 -07:00
