7 Commits

Author	SHA1	Message	Date
Soumith Chintala	b5f7720ab9	docstrings for container and batchnorm	2016-09-16 05:31:36 -04: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
Sam Gross	f45213a276	Simplify nn.Container and nn.Sequential ... - nn.Container.modules is just a python list and used by nn.Sequential - Every module in nn.Sequential has a name. This fixes Module.type() - nn.Sequential constructor accepts either a list or an OrderedDict. With a list, the modules are named "0", "1", "2", ...	2016-08-31 11:15:31 -07:00
Adam Paszke	ea93fb7ac0	Add more nn modules	2016-08-23 19:15:21 -07:00
Adam Paszke	7bcb2a4081	Initial optim version	2016-08-23 19:03:30 -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