search

rdevon / cortex

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

fixing convnet nonlin #185

Closed rdevon closed 6 years ago

rdevon commented 6 years ago

Small fix to convnet. Also option to remove batch norm last layer.