#!/usr/bin/env sh
export MXNET_CUDNN_AUTOTUNE_DEFAULT=1

## train resnet-18 with channel gating
python -u train_resnet480.py --data-dir 'path to imagenet' --data-type imagenet --depth 18 \
       --batch-size=240 --lr=0.1 --gpus=0,1,2 --prune=1 --monitor=5000 --log='resnet-cg' \
       --quantize-w=False --quantize-act=False --groups=8
