https://github.com/bkj/basenet

Pytorch NN helpers
https://github.com/bkj/basenet

computer-vision machine-learning neural-network nlp pytorch

Last synced: about 1 year ago
JSON representation

Pytorch NN helpers

• Host: GitHub
• URL: https://github.com/bkj/basenet
• Owner: bkj
• Created: 2018-02-14T16:59:50.000Z (over 8 years ago)
• Default Branch: master
• Last Pushed: 2024-05-03T19:38:33.000Z (about 2 years ago)
• Last Synced: 2024-05-03T20:58:28.699Z (about 2 years ago)
• Topics: computer-vision, machine-learning, neural-network, nlp, pytorch
• Language: Python
• Homepage:
• Size: 122 KB
• Stars: 20
• Watchers: 4
• Forks: 10
• Open Issues: 3
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

          
#### 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

```