Skip to content

jsbcannell/basenet

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

72 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

basenet

Classes wrapping basic pytorch functionality.

  • Train for an epoch
  • Eval for an epoch
  • Predict
  • Learning rate schedules
Installation
conda create -n basenet4_env python=3.6 pip -y
source activate basenet4_env

pip install -r requirements.txt
conda install -y pytorch torchvision cuda90 -c pytorch
python setup.py install

Examples

cd examples/cifar
./cifar10.sh

About

Pytorch NN helpers

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%