MathGaron / pytorch_toolbox

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

Features/clean loop #11

Closed MathGaron closed 7 years ago

MathGaron commented 7 years ago

This will change the way your network receives its inputs instead of forward(list_of_input) it will be forward(input1, input2, input3)

Also, I made some fix with the weird inconsistencies with parameters as addressed in issue #6