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.

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

[FEATURE] Support shufflenet #1933

Open
flywheel1412 opened this issue Aug 28, 2023 · 3 comments
Open

[FEATURE] Support shufflenet #1933

flywheel1412 opened this issue Aug 28, 2023 · 3 comments
Labels
enhancement New feature or request

Comments

@flywheel1412
Copy link

flywheel1412 commented Aug 28, 2023

tkx for your wonderful work, I found the timm is not support shufflenet v1 and v2 yet, there is any plan to support it? and here is the official repository and related papers.

@flywheel1412 flywheel1412 added the enhancement New feature or request label Aug 28, 2023
@rwightman
Copy link
Collaborator

@flywheel1412 it's never been a priority since they didn't seem particularly stronger or better than other mobile models here already (mobilenet-v2/v3, hardcorenas, tinynet, lcnet, ghostnet, etc). Could likely be supported via efficientnet/mobilenetv3 backbones with additional block type(s) and the extra SE in head... GhostNet is also a very similar model layout that'd be a template to follow... open to well tested full impl with features_only, etc working. But hasn't been a priority for me to tackle.

@flywheel1412
Copy link
Author

tkx for your reply, where cloud i found the benchmark table of timm's models? which cloud be a better guide for everyone to choose models

@rwightman
Copy link
Collaborator

@flywheel1412 the last batch of results are all in the csv files here: https://github.com/huggingface/pytorch-image-models/tree/main/results

benchmark-* are the per architecture inference and train throughput numbers, and the results-* are the per-weight instance accuracy numbers for imagenet and other test sets. You need to join them on the architecture (portion before the .) to get a per result score to plot a pareto curve like I've done in the past https://twitter.com/wightmanr/status/1463684184912711689

Latest GhostNet-V2, RepGhostNet, MobileOne models that I just added are not in the results yet... probably another week or two as it takes a while to run these days.

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

No branches or pull requests

2 participants