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
7023d556b582f2b43d5ab8ad4fc73cd0b3efea0b
AMD-SHARK-Studio/tank
History
Chi_Liu 7023d556b5 Add Debug log of torch_model_blacklist.txt (#242) 
* Add debug log of torch_model_blacklist.txt

* Add make_fx for torch model

* Update torch_model_blacklists.txt

* Add some Xfails
2022-08-09 17:54:02 +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
Add Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +05:30
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
Add Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +05:30
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
Add Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +05:30
cartoongan
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
convbert-base-turkish-cased_tf
Add Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +05:30
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
Add Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +05:30
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
Add Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +05:30
facebook_convnext-tiny-224_tf
Add Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +05:30
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
Add Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +05:30
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
Add Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +05:30
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
Add Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +05:30
mobilebert-uncased_torch
Add Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +05:30
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 Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +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
Add Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +05:30
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 Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +05:30
rembert_tf
Move tf tests to tank/ (#223)
2022-07-29 09:34:50 -07:00
resnet18_torch
Add Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +05:30
resnet50_torch
Add Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +05:30
resnet101_torch
Add Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +05:30
resnet_50_224_int8
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
roberta-base_tf
Add Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +05:30
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
Add Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +05:30
xlm-roberta-base_tf
Add Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +05:30
__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 Debug log of torch_model_blacklist.txt (#242)
2022-08-09 17:54:02 +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