Skip to content

Conversation

@HSShin0
Copy link
Contributor

@HSShin0 HSShin0 commented Mar 30, 2021

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}

HSShin0 added 2 commits March 30, 2021 17:44
- Can test pruned model using `--test-weight` option in `prune.py`
- Formatting
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants