Commit Graph

4 Commits

Author SHA1 Message Date
fb39971464 Add more modules to nn 2016-09-14 11:05:56 -07:00
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
959304bc0d Make BatchNorm2d inherit from BatchNorm 2016-08-30 13:25:01 -07:00
ea93fb7ac0 Add more nn modules 2016-08-23 19:15:21 -07:00