pytorch training tools
Project description
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
pip install -e .
Examples
cd examples/cifar
./run.sh
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
basenet-0.1.0.tar.gz
(10.0 kB
view hashes)