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

## train resnet-18 with channel gating
python -u train_resnet256.py --data-dir 'path to imagenet' --data-type imagenet --depth 18 \
       --batch-size 240 --gpus=0,1,2 --prune=1 --aug-level=1 --groups=8\
       --monitor 5000 --model-load-epoch=100 --lr=0.001 --wd=0 --log='resnet-cg'

