This is a Tool package for PyTorch.
DataTools
contains tools for easy data managementLogTools
contains tools for easy loggingTorchNet
contains modules and some classical Models for Image to Image or ClassificationFunctions
contains functions that useful for pre-processing and testing Docs are available in each folders, check it before first use.
Download the repository, and put it at the root of your project:
- Your Project Folder
|
|- TorchTools
|
|- Your Package
|
|- your python files
|
.
.
.
|
|- python files
Import TorchTools directly in your python files
The functions or class start with _
or __
are Internal implementation, I don't recommend you to use them directly unless you know how it works.
Create issue if you want some functions or tools. Bug reports are NEEDED.
- 18.01.12 First commit