Skip to content

MathGaron/pytorch_toolbox

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

98 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pytorch_toolbox

General tools for using pytorch :

  • Reduce the need for boilerplate code
  • Visualization tools

The main goal if this simple package is to remove the boilerplate code needed for training the network and leave flexibility to the user.

See the Cat vs Dog example.

Dependencies

conda install -c anaconda pyyaml

conda install -c conda-forge tqdm

conda install pytorch torchvision -c soumith

conda install -c anaconda scikit-image

About

Boiler plate code for pytorch. Train/Validation loops, visualization etc. For research.

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •  

Languages