amc

amc copied to clipboard

Hi, thanks for your great work. I want to compare the inference time after compressing MobileNet. I measured with the batch size set to 1 with the code snippet below,...

re-ag

Can the backbone pruned from this code be used for target detection tasks? Or does it mean that this method can compress the task of target detection? @tonylins

APeiZou

Is there a weight file for MobilenetV2?

2

APeiZou

I use the pretrained model, but the result is too bad 

GuoQuanhao

Hello, thanks for your work. I modified your code to prune 3x3 conv layers using im2col in order to prune VGG16. The result is, about 40% accuracy loss from baseline...

capitaso

implement NxN conv (N>1)

22

Hello, I implemented NxN (N>1) convolution case in AMC. You can run test with VGG16 model as follows. bash ./scripts/search_vgg16_0.5flops.sh

capitaso

Compression results for ResNet on ImageNet

2

@zhijian-liu Hi, have you compressed ResNet on ImageNet using AMC, Could you share your result?

xingyueye

The linearRegression for resnet-50 is to slow and reward is low.

3

Hi all, I use the im2col to implement the conv3*3 linearRegression, but the prune speed is slow, it takes about 2 days to finish 800 steps train. The reward is...

aliceyayunji

The linearRegression.fit() is too slow.

1

https://github.com/mit-han-lab/amc/blob/040d83fe9f0288556c8afe363c190e763425a6fe/lib/utils.py#L130

zeng-hello-world

Frist, thank you for your brilliant work! I am wondering that after 2400 episodes of training, how could I reuse the RL model? The --resume argument doesn't work. The load_weights...

6zc