github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00
Code Issues Packages Projects Releases Wiki Activity
Files
3514822cac49a09dda181139a164e0014cfdcfbe
AMD-SHARK-Studio/tank
History
Ean Garvey 3514822cac Improvements to pytest benchmarks. (#267) 
* Add ONNX env var flags for venv setup.

* Setup arguments for ONNX benchmarking via pytest.

* Enable ONNX benchmarking on MiniLM via pytest (experimental)

* Fix sequence lengths to 128 for TF model creation and fix issue with benchmarks.

* Disable CI CPU benchmarks on A100, change some default args.

* add xfails for roberta TF model tests on GPU.
2022-08-17 02:29:48 -05:00
..
albert_lite_base
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
albert-base-v2_tf
Revert "Add Debug log of torch_model_blacklist.txt (#242)" (#249)
2022-08-09 10:23:14 -07:00
albert-base-v2_torch
Move torch tests up to /tank (#234)
2022-08-03 10:50:53 -07:00
alexnet_torch
Move torch tests up to /tank (#234)
2022-08-03 10:50:53 -07:00
arbitrary-image-stylization-v1-256
Test tflite quantization_info fail on cloud (#228)
2022-08-01 18:21:04 -07:00
bert_fine_tuning
fine tune with shark (#211)
2022-08-04 13:14:57 -05:00
bert-base-cased_torch
Add more torch hg model tests (#238)
2022-08-03 18:00:04 -07:00
bert-base-uncased_tf
Improvements to pytest benchmarks. (#267)
2022-08-17 02:29:48 -05:00
bert-base-uncased_torch
Improvements to pytest benchmarks. (#267)
2022-08-17 02:29:48 -05:00
birds_V1
Test tflite quantization_info fail on cloud (#228)
2022-08-01 18:21:04 -07:00
camembert-base_tf
Revert "Add Debug log of torch_model_blacklist.txt (#242)" (#249)
2022-08-09 10:23:14 -07:00
cartoongan
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
convbert-base-turkish-cased_tf
Revert "Add Debug log of torch_model_blacklist.txt (#242)" (#249)
2022-08-09 10:23:14 -07:00
deberta-base_tf
Move tf tests to tank/ (#223)
2022-07-29 09:34:50 -07:00
deeplabv3
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
densenet
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
distilbert-base-uncased_tf
Revert "Add Debug log of torch_model_blacklist.txt (#242)" (#249)
2022-08-09 10:23:14 -07:00
distilbert-base-uncased_torch
Move torch tests up to /tank (#234)
2022-08-03 10:50:53 -07:00
efficientnet_224_fp32
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
efficientnet_lite0_fp32_2
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
efficientnet_lite0_int8_2
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
electra-small-discriminator_tf
Revert "Add Debug log of torch_model_blacklist.txt (#242)" (#249)
2022-08-09 10:23:14 -07:00
facebook_convnext-tiny-224_tf
Revert "Add Debug log of torch_model_blacklist.txt (#242)" (#249)
2022-08-09 10:23:14 -07:00
funnel-transformer_tf
Move tf tests to tank/ (#223)
2022-07-29 09:34:50 -07:00
google_vit-base-patch16-224_tf
Add tf image classification auto model tests (#216)
2022-07-27 18:38:13 -07:00
gpt2-64
Test tflite quantization_info fail on cloud (#228)
2022-08-01 18:21:04 -07:00
inception_v4_299_fp32
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
inception_v4_299_uint8
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
layoutlm-base-uncased_tf
Revert "Add Debug log of torch_model_blacklist.txt (#242)" (#249)
2022-08-09 10:23:14 -07:00
longformer-base-4096_tf
Move tf tests to tank/ (#223)
2022-07-29 09:34:50 -07:00
midas
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
MiniLM-L12-H384-uncased
Improvements to pytest benchmarks. (#267)
2022-08-17 02:29:48 -05:00
MiniLM-L12-H384-uncased_torch
Improvements to pytest benchmarks. (#267)
2022-08-17 02:29:48 -05:00
mnasnet_1.0_224
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
mobilebert
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
mobilebert-baseline-tf2-float
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
mobilebert-baseline-tf2-quant
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
mobilebert-edgetpu-s-float
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
mobilebert-edgetpu-s-quant
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
mobilebert-uncased_tf
Revert "Add Debug log of torch_model_blacklist.txt (#242)" (#249)
2022-08-09 10:23:14 -07:00
mobilebert-uncased_torch
Revert "Add Debug log of torch_model_blacklist.txt (#242)" (#249)
2022-08-09 10:23:14 -07:00
mobilenet_v1_224_1.0_float
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
mobilenet_v1_224_1.0_uint8
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
mobilenet_v2_1.0_224
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
mobilenet_v2_1.00_224_int8
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
mobilenet_v2_224_1.0_uint8
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
mobilenet_v3_small_torch
Revert "Add Debug log of torch_model_blacklist.txt (#242)" (#249)
2022-08-09 10:23:14 -07:00
mobilenet_v3-large_224_1.0_float
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
mobilenet_v3-large_224_1.0_uint8
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
mobilenet_v3.5multiavg_1.00_224_int8
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
mpnet-base_tf
Revert "Add Debug log of torch_model_blacklist.txt (#242)" (#249)
2022-08-09 10:23:14 -07:00
multi_person_mobilenet_v1_075_float
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
nasnet
Test tflite quantization_info fail on cloud (#228)
2022-08-01 18:21:04 -07:00
pytorch
Revert "Add Debug log of torch_model_blacklist.txt (#242)" (#249)
2022-08-09 10:23:14 -07:00
rembert_tf
Move tf tests to tank/ (#223)
2022-07-29 09:34:50 -07:00
resnet18_torch
Revert "Add Debug log of torch_model_blacklist.txt (#242)" (#249)
2022-08-09 10:23:14 -07:00
resnet50
Improvements to pytest benchmarks. (#267)
2022-08-17 02:29:48 -05:00
resnet50_torch
Revert "Add Debug log of torch_model_blacklist.txt (#242)" (#249)
2022-08-09 10:23:14 -07:00
resnet101_torch
Revert "Add Debug log of torch_model_blacklist.txt (#242)" (#249)
2022-08-09 10:23:14 -07:00
resnet_50_224_int8
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
roberta-base_tf
Improvements to pytest benchmarks. (#267)
2022-08-17 02:29:48 -05:00
squeezenet
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
squeezenet1_0_torch
Move torch tests up to /tank (#234)
2022-08-03 10:50:53 -07:00
ssd_mobilenet_v1_320_1.0_float
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
ssd_mobilenet_v1_320_1.0_uint8
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
ssd_mobilenet_v2_face_quant
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
ssd_spaghettinet_edgetpu_large
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
ssd_spaghettinet_edgetpu_large_uint8
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
tapas-base_tf
Move tf tests to tank/ (#223)
2022-07-29 09:34:50 -07:00
tf
Add benchmarks to MHLO miniLM and resnet50 and add dialect, num_iterations (#264)
2022-08-16 13:55:40 -05:00
tflite
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
tiny-random-flaubert_tf
Move tf tests to tank/ (#223)
2022-07-29 09:34:50 -07:00
wide_resnet50_2_torch
Revert "Add Debug log of torch_model_blacklist.txt (#242)" (#249)
2022-08-09 10:23:14 -07:00
xlm-roberta-base_tf
Improvements to pytest benchmarks. (#267)
2022-08-17 02:29:48 -05:00
__init__.py
Add __init__.py for tank/
2022-06-12 21:53:59 -07:00
model_utils_tf.py
Improvements to pytest benchmarks. (#267)
2022-08-17 02:29:48 -05:00
model_utils.py
Add benchmarks to MHLO miniLM and resnet50 and add dialect, num_iterations (#264)
2022-08-16 13:55:40 -05:00
README.md
fine tune with shark (#211)
2022-08-04 13:14:57 -05:00

README.md

To run the fine tuning example, from the root SHARK directory, run:

IMPORTER=1 ./setup_venv.sh
source shark.venv/bin/activate
pip install jupyter tf-models-nightly tf-datasets
jupyter-notebook

if running from a google vm, you can view jupyter notebooks on your local system with:

gcloud compute ssh <YOUR_INSTANCE_DETAILS> --ssh-flag="-N -L localhost:8888:localhost:8888"
Reference in New Issue View Git Blame Copy Permalink