Created by: wanghaoshuang

python train.py \
--model "ResNet34" \
--criterion "geometry_median" \
--pruned_ratio=0.3125

2020-04-23 16:29:22,737-INFO: FLOPs before pruning: 69854464.0
2020-04-23 16:29:24,723-INFO: FLOPs after pruning: 33773584.0