search

MathGaron / pytorch_toolbox

Boiler plate code for pytorch. Train/Validation loops, visualization etc. For research.
MIT License
10 stars 3 forks source link

Feature/probe #30

Closed MathGaron closed 5 years ago

MathGaron commented 5 years ago

Mostly some features to keep activation in a dict (nothing fancy!)