search

EkdeepSLubana / OrthoReg

MIT License
9 stars 2 forks source link

Iterative pruning example not executing #2

Closed piyush-das closed 3 years ago

piyush-das commented 3 years ago

Hi,

Although the iterative pruning segment mentions the following command with pretrained=False python main.py --model=vgg --pretrained=False --pruning_type=orthoreg --prune_percent=50 --n_rounds=2

On executing it we get the following error FileNotFoundError: [Errno 2] No such file or directory: 'pretrained/iterative/vgg.pth'

Please let me know if I am missing something.

Thanks

EkdeepSLubana commented 3 years ago

Hi!

Just made an update. Since it's been a while, I can't recall why, but there was a conditional statement in the scratch training process. It's highly likely that I included the conditional because I was running some experiments and forgot to remove it. Let me know if there are more issues.

piyush-das commented 3 years ago

Thanks, for the update. I'll be sure to let you know if I have any more findings.