Eric-mingjie / rethinking-network-pruning

Rethinking the Value of Network Pruning (Pytorch) (ICLR 2019)
MIT License
1.51k stars 293 forks source link

Error while running l1-norm-pruning on Windows machine #57

Open mitrapallavi11 opened 1 year ago

mitrapallavi11 commented 1 year ago

Hi,

I am trying to run the code for l1-norm-pruning on windows machine. But I am getting an error due to multiprocessing failure as follows:

RuntimeError: An attempt has been made to start a new process before the current process has finished its bootstrapping phase.

This probably means that you are not using fork to start your child processes and you have forgotten to use the proper idiom in the main module:

if name == 'main': freeze_support() ...

The "freeze_support()" line can be omitted if the program is not going to be frozen to produce an executable.

Any solution how to mitigate that error?

Eric-mingjie commented 1 year ago

can you point me which file are you referring to here?

mitrapallavi11 commented 1 year ago

training VGG-16

python main.py --dataset cifar10 --arch vgg --depth 16

Eric-mingjie commented 1 year ago

can you point to which file? I don't think we have the function call to freeze_support anywhere in the code.