search

rdevon / cortex

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

Dev #172

Closed rdevon closed 6 years ago

rdevon commented 6 years ago

Various bug fixes and enhancements:

visualize no longer ignores errors. Added STL options. Added option to not average in GAN scores Added option to ignore nonlinearity on last conv layer Updated resnets (needs to be tested) There was a cuda error with some nonlinearities (PReLU). Added a fix, but probably cleaner solution exists.

rdevon commented 6 years ago

Fixed the errors. They were related to both the visualization changes and that tests were being done on mixed cpu /gpu.