kws02352/Model_Compression_Boostcamp

Model Compression : Pruning & decomposition

	MACs	F1	Competition Score
Before	1688940.0	0.6206	0.4005
After	1083210.0	0.6149	0.3431

Test_Demo

Before

python test_demo.py --customized_model False --eval True

After

python test_demo.py --customized_model True --eval True

Envrionment

pytorch '1.7.1+cu101'
albumentations
sklearn
ptflops

Directory

Model Compression/
├──input/
|  └── data/
|      ├── train/
|          ├── trainImg00001.jpg
|          ├── ...
|          └── trainImg22640.jpg
|      └── val/
|          ├── valImg0001.jpg
|          ├── ...
|          └── valImg8816.jpg
├──pretrained/
│  ├── ShuffleNet_final.pt
│  └── shufflenet_g3_wd4.pth
├──solution/
|  ├── dataloader.py
|  ├── ...
|  └── utils.py
└──test_demo.py