L-OBS

L-OBS copied to clipboard

Some question about the pytorch implementation

3

Hi, I modified your code for pruning a ResNet18 model trained on CIFAR10 datasets, I was confused with the function `adjust_mean_var` in `utils.py`. In your code you put model in...

chamboin

@csyhhu Hello, I am an undergraduate. I need to use the most basic OBS algorithm to write my graduation thesis. Only you have written the code about OBS on the...

sunhongfei1997

Some questions about re-implementing L-OBS on Lenet5

6

Hi, Thank you very much for sharing your code. I am trying to re-implementing your L-OBS algorithm for the purpose of learning. I have successfully used this algorithm in a...

huiouyang16

Difficulty in reproducing NIPS'17 results for AlexNet using PyTorch code

9

Hi Shangyu, Thanks a lot for sharing PyTorch code for applying LOBS on various ImageNet CNNs. I could run the code perfectly after a couple of minor error/syntax corrections required...

adpatil2

bug in prune-weights-ResNet18.py?

1

Hi, I was looking into the ResNet18 pytorch code, and noticed the following sequence: ```# Specify layer type, C for convolution, F for fully-connected if layer_name == 'fc': layer_type =...

dalistarh

I was running your code for mnist on tensorflow. I commented out computing Hessians and edge cuts (lines 219-248), So I load your precomputed weights after pruning. This gives me...

rahimentezari

Tensorflow-Prune on GPU

1

Is there a way to run your tensorflow code for Lenet 300-100 on GPU? like as you did for computing hessian for Alexnet on GPU. Pruning on CPU takes very...

rahimentezari

AlexNet - TinyImagenet

4

First of all thanks for sharing the code. I used your pytorch code for Alexnet on Tiny ImageNet. When I tried to retrain the network after pruning, I got the...

rahimentezari

ValueError: GraphDef cannot be larger than 2GB.

1

Thank you very much for sharing the code publicly. I tweaked the code to run it on GPU without requiring much memory. (can send a PR) When I ran calculate_hessian_inverse.py...

Kaushalya

Found some error when use L-OBS to prune deep cnn(alexnet, vgg, etc)

3

Hi Shangyu, Thanks for sharing the code! After reading your paper, I do some experiments. I found the code at L-OBS/lenet300-100 do a good jop when prune lenet300-100(some small bug...

dsfour