Files

benchmarks

Failed to load latest commit information.

Cannot retrieve latest commit at this time.

Name		Name	Last commit message	Last commit date
parent directory ..
configs		configs
models		models
README.md		README.md
__init__.py		__init__.py
benchmark.py		benchmark.py
requirements.txt		requirements.txt
train.py		train.py
utils.py		utils.py

README.md

Image Classification Benchmarks

Setup

Install octconv:
- (Option 1) From pip:
```
pip install octconv
```
- (Option 2) Locally:
```
pip install -e ..   
```
Install remaining requirements
```
  pip install -r requirements.txt
```

Training

Single GPU

python train.py -c configs/cifar10/oct-resnet20.yml --device cuda:0

Multi-GPU

NGPUS=4; python -m torch.distributed.launch --nproc_per_node ${NGPUS} train.py -c configs/cifar10/oct-resnet20.yml