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 more vision models from mlagility #614

Merged
merged 75 commits into from
Jun 21, 2023
Merged

Add more vision models from mlagility #614

merged 75 commits into from
Jun 21, 2023

Conversation

jcwchen
Copy link
Member

@jcwchen jcwchen commented Jun 20, 2023 •
edited
Loading

Follow-up PR of #605 -- add more vision models from mlagility. They are all existing models under vision, but with higher opset_version 18.

Added 14 models from torch_hub and torchvision -- now have 17 models in total:
maskrcnn_resnet50_fpn, ssd300_vgg16, mobilenet_v2, mobilenet_v3, resnet18, resnet34, resnet101, vgg16, vgg16_bn, vgg19, bgg19_bn, densenset121, inception_v2, googlenet.

cc @ramkrishna2910

@jcwchen
apply mlagility to create ONNX model in CI
a5a82ef 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
correct path
9db5fb2 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
which benchit
65c5d7e 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
cd mlagility
d913609 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
pip install -e .
ccbaf72 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
add models
b24081d 
Signed-off-by: Chun-Wei Chen <[email protected]>
@jcwchen
../mlagility_models/alexnet_torch_hub_2891f54c/onnx/
acf2326 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
git diff --exit-code -- .
2731ca4 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
add opset 17 models
398b63d 
Signed-off-by: Chun-Wei Chen <[email protected]>
@jcwchen
add faster-rcnn in CIs
f6c54a1 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
cd
0c06ef3 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
use 18 instead of 17
ac5f5ec 
Signed-off-by: Chun-Wei Chen <[email protected]>
@jcwchen
Merge branch 'jcw/add-mlagility' of https://github.com/jcwchen/models …
05c495d 
…into jcw/add-mlagility
@jcwchen
test opset_version 18
1a583ec 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
git diff subdirectory
1f1db90 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
torch==2.0.0 torchvision==0.15.1
500b542 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
run_mlagility.py
ab889ae 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
use better path to store .onnx
0380b2f 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
print exception
6ff1014 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
copy
db69afb 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
+base_name
a0f0339 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
correct path from mlagility
dd41d77 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
add another way to upload new model
2795d20 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
validate_model
37cf55e 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
import os.path as osp
cea6c52 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
correct path
19f6bc3 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
from pathlib import Path
934765a 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
pip install -r
a6f9618 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
[]
fa5d290 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
fix
00e4e01 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
mlagility_models_dir
d7d3d27 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
new config
7af0369 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
add 3 more models
355c6c3 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
add config
4dae47a 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
remove bert and gpt2 for now
0e1a570 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
clean path
0ef4ef4 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
Add more vision models from mlagility
4bc2cdc 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
remove models in wrong dir
fe74ba2 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
add new models in config
67dd03b 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
Merge branch 'new-models' into jcw/more-vision-models
4423c93 
Signed-off-by: Chun-Wei Chen <[email protected]>
@jcwchen
find_model_hash_name
6a4ba08 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
switch order
7b752ff 
Signed-off-by: jcwchen <[email protected]>
@ramkrishna2910 ramkrishna2910 self-assigned this Jun 20, 2023
@jcwchen
requirements
7195930 
Signed-off-by: jcwchen <[email protected]>
@ramkrishna2910 ramkrishna2910 removed their assignment Jun 20, 2023
@jcwchen
check=True
139fcc9 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
modified models
fcb1b1c 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
fix config
b2210f0 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
rename
8a591a8 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
add missing config
657c375 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
add skip and update args
77f4a3c 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
create test_data_set on the fly
1337fe8 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
remove maskrcnn v2
e1f7704 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
use final_model_dir
8b0fb00 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
model_hash_name + base_name
902d751 
Signed-off-by: jcwchen <[email protected]>
@jcwchen
add maskrcnn
b8e54db 
Signed-off-by: jcwchen <[email protected]>
@jcwchen jcwchen merged commit 02ab9d0 into onnx:new-models Jun 21, 2023
@jcwchen jcwchen deleted the jcw/more-vision-models branch June 21, 2023 22:00
@jcwchen jcwchen mentioned this pull request Jun 23, 2023
Open
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@ramkrishna2910 ramkrishna2910 ramkrishna2910 approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@jcwchen @ramkrishna2910