Soumith Chintala
513d902df1
adding __repr__ for nn
2016-11-07 16:17:40 -05:00
Adam Paszke
b4f4cca875
Rename training and evaluation methods
2016-10-30 00:16:06 +02:00
Adam Paszke
3cbe66ba8c
Change requires_grad default to False
2016-10-05 08:46:34 -07:00
soumith
d92b7da733
fix documentation to not use forward
2016-09-30 09:49:30 -07:00
Adam Paszke
c8a4734b97
Add RReLU to both nn packages
2016-09-29 11:33:34 -07:00
Adam Paszke
f9d25e8e72
Refactor nn (require specifying parameters explicitly)
2016-09-27 15:22:26 -07:00
Soumith Chintala
b5f7720ab9
docstrings for container and batchnorm
2016-09-16 05:31:36 -04:00
soumith
a0a2d9885a
adding docstrings for activation functions
2016-09-16 03:40:24 -04: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
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
e055ffbdc7
Add nn
2016-08-19 14:56:55 -07:00