rdevon / cortex

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

Adding variational autoencoders with new format #76

Closed rdevon closed 6 years ago

rdevon commented 6 years ago

Added VAE to cortex as well as numerous other functionality.

VAEs are the first model that uses plugins from other modules. Notably, the simple classifier now added to cortex/built_ins/models/classifier.py is used in VAE ModelPlugin.

In order to do this, I needed to added more referencing to the kwarg passing, so that the user could have control what parameters were being passed to the build functions. This may have added some difficult-to-interpret behavior in the back-end, which needs to be cleaned up in the future to make development clear.