Helper functions and scripts for pytorch
- turn into classes
- add docstrings
- make into a pypi package (ver 0.1a)
- add support for custom predict function
- add metrics parser to existing metrics
- add more metrics to the wrapper
- plotting capabilities
- find another way to impl logger (maybe decorator)