rdevon / cortex

A machine learning library for PyTorch
BSD 3-Clause "New" or "Revised" License
92 stars 25 forks source link

Base network #151

Closed rdevon closed 6 years ago

rdevon commented 6 years ago

Adding a base network that is the parent of all built-in networks. Moved other network subclassing this. This includes a lot of redundant code related to passing arguments, nonlinearities, and forward method.

Also removed some unused networks (MNIST).