model_compression
model_compression copied to clipboard
Add test option for pruned model
I added an option for testing pruned model.
Example:
python prune.py --resume {resume_dir_name} --test-weight {resume_dir_name/x_xx_xx_pruned/xx_xx_xx.pth.tar}