packnet
packnet copied to clipboard
Code for PackNet: Adding Multiple Tasks to a Single Network by Iterative Pruning
I am new in this task and read your code briefly。 Does line 137 in src/prune.py : ' self.current_masks = self.previous_masks' should be 'self.current_masks = mask'?
Hi @arunmallya : Running Platform: _GoogleColab_ Environment: _Python3.6 torch==0.2.0.post3 torchvision==0.1.9 torchnet, tqdm_ I have changed "raw_input()" to "input()" in **network.py**. When I try to run the program, the following error...
Hello Please, I have a problem concerning the optimization of YOLOX model by Pytorch Pruning (Global & Local) I got the same inference time after the optimization, I don't understand...