Eric-mingjie / rethinking-network-pruning

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

VGG-16 and ResNet-50 from Pytorch Model Zoo Not found #56

Open mitrapallavi11 opened 1 year ago

mitrapallavi11 commented 1 year ago

In weight pruning with ImageNet dataset, the sources of baselines VGG-16 and RESNET-50 are not found. From where the pretrained model can be downloaded or training can be run?

Eric-mingjie commented 1 year ago

I think it is mostly due to pytorch model zoos being updated. Maybe you can find checkpoints for base model in timm.

mitrapallavi11 commented 1 year ago

I think, this Is related to Window’s multiprocessing problem. I changed num_workers = 0 and it solved the problem. @.***

Internal From: Eric-mingjie @.> Sent: Thursday, 11 May 2023 9:14 am To: Eric-mingjie/rethinking-network-pruning @.> Cc: Mitra, Pallavi (uif62213) @.>; Author @.> Subject: Re: [Eric-mingjie/rethinking-network-pruning] VGG-16 and ResNet-50 from Pytorch Model Zoo Not found (Issue #56)

CAUTION: This is an external email. Do not click or open any attachments unless you recognize the sender and know that the content is safe. (http://links.conti.de/mailcheck)

I think it is mostly due to pytorch model zoos being updated. Maybe you can find checkpoints for base model in timmhttps://github.com/huggingface/pytorch-image-models/tree/main.

— Reply to this email directly, view it on GitHubhttps://github.com/Eric-mingjie/rethinking-network-pruning/issues/56#issuecomment-1543458095, or unsubscribehttps://github.com/notifications/unsubscribe-auth/A7EBMR5SWFE5ZCL7CYYYCB3XFSGTLANCNFSM6AAAAAAX2ELCK4. You are receiving this because you authored the thread.Message ID: @.***>