Crashing while creating file `frozen500.pickle`

ethanhe42 / channel-pruning

Channel Pruning for Accelerating Very Deep Neural Networks (ICCV'17)

https://arxiv.org/abs/1707.06168

MIT License

1.08k stars 310 forks source link

Closed ddfabbro closed 6 years ago

ddfabbro commented 6 years ago

I'm following the example found in the README, but during step 2

python3 train.py -action c3 -caffe 0

my computer crashes.

The last generated file is frozen500.pickle with size of 5Gb.

What is causing this? Maybe I could lower memory usage?

ethanhe42 commented 6 years ago

You can sample less patches if you don't have enough memory. set nBatches in lib/cfgs.py to like 100, 300. But the accuracy is not guaranteed.