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
9035a2eed3bee1e8afc5afb82690f5dcceadd775
AMD-SHARK-Studio/tank
History
Ean Garvey 9035a2eed3 Add --local_tank_cache flag and update requirements. (#368) 
* Add --local_tank_cache flag and update requirements.

* Update requirements-importer.txt
2022-09-28 03:02:59 -05:00
..
albert_lite_base
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -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-uncased_tosa_torch
Add bert_tosa script
2022-09-23 10:52:06 +05:30
birds_V1
Test tflite quantization_info fail on cloud (#228)
2022-08-01 18:21:04 -07:00
cartoongan
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
deberta-base_tf
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05: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
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
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
midas
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -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
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-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
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
update xfails for PyTorch DistilBERT (#355)
2022-09-24 14:53:20 -05:00
rembert_tf
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05:00
resnet_50_224_int8
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
squeezenet
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -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
Remove "gpu" device alias and migrate to using "cuda" for NVIDIA GPU. (#325)
2022-09-13 01:16:56 -05:00
tf
Fix sourcing for canonical MiniLM shark_tank model artifacts. (#278)
2022-08-17 23:03:47 -05:00
tflite
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
__init__.py
Add __init__.py for tank/
2022-06-12 21:53:59 -07:00
all_models.csv
creates abstract test case class (#333)
2022-09-20 07:06:38 -07:00
model_metadata.csv
creates abstract test case class (#333)
2022-09-20 07:06:38 -07:00
model_utils_tf.py
creates abstract test case class (#333)
2022-09-20 07:06:38 -07:00
model_utils.py
creates abstract test case class (#333)
2022-09-20 07:06:38 -07:00
README.md
fine tune with shark (#211)
2022-08-04 13:14:57 -05:00
test_models.py
Add --local_tank_cache flag and update requirements. (#368)
2022-09-28 03:02:59 -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