arunmallya / packnet

Code for PackNet: Adding Multiple Tasks to a Single Network by Iterative Pruning
https://arxiv.org/abs/1711.05769
229 stars 40 forks source link

is there any way to upgrade to newer pytorch version #4

Closed pardont closed 2 years ago

pardont commented 3 years ago

hi @arunmallya ,I try to run script ./scripts/run_lwf.sh csf ../checkpoints/imagenet/vgg16.pt 2,3

seems vgg16.pt file worked with pytorch 0.2.0.post3, and can download from https://uofi.box.com/s/zap2p03tnst9dfisad4u0sfupc0y1fxt

is there any way to upgrade to newer pytorch version, such as pytorch 1.6

Thanks

arunmallya commented 3 years ago

Hi, I am not actively maintaining this at the moment. You can try using a newer pytorch version and fixing issues as they come along. If you can get it to work, please feel free to send a merge request.