Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add shufflenetv2 1.5 and 2.0 weights #5906

Merged
merged 16 commits into from
Apr 28, 2022
Merged

Add shufflenetv2 1.5 and 2.0 weights #5906

Changes from 1 commit
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
d89f57a
Add shufflenetv2 1.5 and 2.0 weights
YosuaMichael Apr 27, 2022
c0516e7
Update recipe
YosuaMichael Apr 27, 2022
de2cf45
Add to docs
YosuaMichael Apr 27, 2022
5487b23
Use resize_size=232 for eval and update the result
YosuaMichael Apr 27, 2022
906fa30
Add quantized shufflenetv2 large
YosuaMichael Apr 27, 2022
de5d638
Update docs and readme
YosuaMichael Apr 27, 2022
7706d08
Format with ufmt
YosuaMichael Apr 27, 2022
06836b5
Add to hubconf.py
YosuaMichael Apr 27, 2022
b738c7d
Update readme for classification reference
YosuaMichael Apr 28, 2022
687498a
Merge branch 'main' into models/add-shufflenetv2-large
YosuaMichael Apr 28, 2022
de5a3b7
Fix reference classification readme
YosuaMichael Apr 28, 2022
3a2eec4
Merge branch 'main' into models/add-shufflenetv2-large
YosuaMichael Apr 28, 2022
dd97bc3
Fix typo on readme
YosuaMichael Apr 28, 2022
ef2dbc5
Merge branch 'main' into models/add-shufflenetv2-large
YosuaMichael Apr 28, 2022
c5551d7
Update reference/classification/readme
YosuaMichael Apr 28, 2022
691d8f6
Merge branch 'main' into models/add-shufflenetv2-large
YosuaMichael Apr 28, 2022
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
Prev Previous commit
Next Next commit
Update reference/classification/readme
  • Loading branch information
@YosuaMichael
YosuaMichael committed Apr 28, 2022
commit c5551d7a63e0283e112d910130c66f32b1343c9e
2 changes: 1 addition & 1 deletion references/classification/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -247,7 +247,7 @@ torchrun --nproc_per_node=8 train.py \
```
Here `$MODEL` is either `shufflenet_v2_x1_5` or `shufflenet_v2_x2_0`.

The models `shufflenet_v2_x0_5` and `shufflenet_v2_x1_0` were contributed by community. See [PR-849](https://github.com/pytorch/vision/pull/849#issuecomment-483391686) for details.
The models `shufflenet_v2_x0_5` and `shufflenet_v2_x1_0` were contributed by the community. See [PR-849](https://github.com/pytorch/vision/pull/849#issuecomment-483391686) for details.


YosuaMichael marked this conversation as resolved.
Show resolved Hide resolved
## Mixed precision training
Expand Down