Skip to content
This repository has been archived by the owner on Sep 18, 2024. It is now read-only.

Update ComperssionUtils Doc #2969

Merged
merged 2 commits into from
Oct 19, 2020
Merged

Update ComperssionUtils Doc #2969

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
f110540
Merge pull request #2959 from microsoft/v1.9
liuzhe-lz Oct 15, 2020
1817648
update title level
Oct 16, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion docs/en_US/Compression/CompressionUtils.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -120,7 +120,7 @@ from nni.compression.torch.utils.mask_conflict import fix_mask_conflict
fixed_mask = fix_mask_conflict('./resnet18_mask', net, data)
```

### Model FLOPs/Parameters Counter
## Model FLOPs/Parameters Counter
We provide a model counter for calculating the model FLOPs and parameters. This counter supports calculating FLOPs/parameters of a normal model without masks, it can also calculates FLOPs/parameters of a model with mask wrappers, which helps users easily check model complexity during model compression on NNI. Note that, for sturctured pruning, we only identify the remained filters according to its mask, which not taking the pruned input channels into consideration, so the calculated FLOPs will be larger than real number (i.e., the number calculated after Model Speedup).

### Usage
Expand Down