github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-25 03:00:12 -04:00
Code Issues Packages Projects Releases Wiki Activity
Files
d934765b1d98b71641d19d847923aa6eca32e5f4
AMD-SHARK-Studio/tank
History
Prashant Kumar d934765b1d Add mobilenet_v3_small torch model to the test_suite. 
-- The model doesn't validate with the correct results on the GPU.
-- The model passes on CPU and levelzero.
-- The static version of the model gets stuck for vulkan.
2022-08-05 14:10:43 +05:30
..
albert_lite_base
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
albert-base-v2_tf
Move tf tests to tank/ (#223)
2022-07-29 09:34:50 -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
Move tf tests to tank/ (#223)
2022-07-29 09:34:50 -07:00
bert-base-uncased_torch
Move torch tests up to /tank (#234)
2022-08-03 10:50:53 -07:00
birds_V1
Test tflite quantization_info fail on cloud (#228)
2022-08-01 18:21:04 -07:00
camembert-base_tf
Move tf tests to tank/ (#223)
2022-07-29 09:34:50 -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
Mark XFail for M1 Vulkan Failures (#235)
2022-08-02 19:56:02 -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
Move tf tests to tank/ (#223)
2022-07-29 09:34:50 -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
Move tf tests to tank/ (#223)
2022-07-29 09:34:50 -07:00
facebook_convnext-tiny-224_tf
Add tf image classification auto model tests (#216)
2022-07-27 18:38:13 -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
Move tf tests to tank/ (#223)
2022-07-29 09:34:50 -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_torch
Move torch tests up to /tank (#234)
2022-08-03 10:50:53 -07: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
Move tf tests to tank/ (#223)
2022-07-29 09:34:50 -07:00
mobilebert-uncased_torch
Move torch tests up to /tank (#234)
2022-08-03 10:50:53 -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
Add mobilenet_v3_small torch model to the test_suite.
2022-08-05 14:10:43 +05:30
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
Move tf tests to tank/ (#223)
2022-07-29 09:34:50 -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
Add mobilenet_v3_small torch model to the test_suite.
2022-08-05 14:10:43 +05:30
rembert_tf
Move tf tests to tank/ (#223)
2022-07-29 09:34:50 -07:00
resnet18_torch
Move torch tests up to /tank (#234)
2022-08-03 10:50:53 -07:00
resnet50_torch
Move torch tests up to /tank (#234)
2022-08-03 10:50:53 -07:00
resnet101_torch
Move torch tests up to /tank (#234)
2022-08-03 10:50:53 -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
Move tf tests to tank/ (#223)
2022-07-29 09:34:50 -07: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 hash of mlir for tf/tflite (#225)
2022-08-01 10:40:49 -07: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
Move torch tests up to /tank (#234)
2022-08-03 10:50:53 -07:00
xlm-roberta-base_tf
Move tf tests to tank/ (#223)
2022-07-29 09:34:50 -07:00
__init__.py
Add __init__.py for tank/
2022-06-12 21:53:59 -07:00
model_utils_tf.py
Add tf image classification auto model tests (#216)
2022-07-27 18:38:13 -07:00
model_utils.py
Add mobilenet_v3_small torch model to the test_suite.
2022-08-05 14:10:43 +05:30
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