search

yiwenguo / Dynamic-Network-Surgery

Caffe implementation for dynamic network surgery.
Other
186 stars 70 forks source link

How to resume the training using solverstate? #24

Closed umadevimcw closed 6 years ago

umadevimcw commented 6 years ago

Hai, I tried to resume the pruning training using solver state. During resumption the mu and std values are zero. Hence pruning is not happening (X factor is always 1 while I am resuming the training). It seems that mu and std values are not stored. Is it possible to resume or storing the mu and std values? How to resume the pruning training using solver state?

umadevimcw commented 6 years ago

I am closing this issue because, I have added support to resume using snapshot file by storing mu and std values in blobs.