MathGaron / pytorch_toolbox

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

add common modules #16

Closed MathGaron closed 7 years ago

MathGaron commented 7 years ago

the example is updated with the new FC and conv2d block which put together to generic stuff to make our code clean

Of course these module have some limitations if we want to use specific parameters, but this definition should be enough most of the time

MathGaron commented 7 years ago

Bump :p