pytorch-pruning

pytorch-pruning copied to clipboard

CUDA out of memory error: while allocating the memory

7

Hi I am working on Tesla K40, 12 GB GPU machine. I am facing this error constantly. If I calculate the required memory for VGG model with respect to the...

jagadeesh09

when running the project, errors occurs and don't know why  maybe someone can help me with this?

wangjunkai1999

Hello, I am new to github and pytorch. I do not know if my question is appropriate. I have cloned the project to google colab and run the command line...

tdd2454

@jacobgil I find there is no limits when using `optimizer.step()`. So the pruned weight will get a gradient and after stepping, it will be no longer **0** which means it...

igo312

I don't see it stated anywhere it is in prune.py

ooodragon94

Farheen-github

torch.load(pruned_model)

7

when i use torch.load() to load pruned model , error happened: AttributeError: 'module' object has no attribute 'ModifiedVGG16Model', anyone meet this problem?

wuzhiyang2016

I trained the vgg and saved the model as pth file. then I load it for pruning some filters of it. the last conv after pruning is not 512 anymore,...

Saharkakavand

why set layer_index=28, filter_index=10? thanks?

zhanghongruiupup

I wonder if this pruning approach can be adapted to net like resnet

alanMachineLeraning