github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-10 06:17:55 -05:00
Code Issues Packages Projects Releases Wiki Activity
Files
4e4c34c717a23ae670a1929b99daee049b80c9a9
SHARK-Studio/tank
History
Ean Garvey 6cf5564c84 Remove "gpu" device alias and migrate to using "cuda" for NVIDIA GPU. (#325) 
* Replace instances of "gpu" alias for devices with "cuda"
2022-09-13 01:16:56 -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
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05:00
albert-base-v2_torch
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05:00
alexnet_torch
Update xfails for torchvision models. (#310)
2022-09-01 13:06:12 -05: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
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05:00
bert-base-uncased_tf
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05:00
bert-base-uncased_torch
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05:00
birds_V1
Test tflite quantization_info fail on cloud (#228)
2022-08-01 18:21:04 -07:00
camembert-base_tf
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05:00
cartoongan
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
convbert-base-turkish-cased_tf
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05: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
distilbert-base-uncased_tf
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05:00
distilbert-base-uncased_torch
Remove "gpu" device alias and migrate to using "cuda" for NVIDIA GPU. (#325)
2022-09-13 01:16:56 -05: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
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05:00
facebook_convnext-tiny-224_tf
Remove "gpu" device alias and migrate to using "cuda" for NVIDIA GPU. (#325)
2022-09-13 01:16:56 -05:00
facebook_deit-small-distilled-patch16-224_torch
Remove "gpu" device alias and migrate to using "cuda" for NVIDIA GPU. (#325)
2022-09-13 01:16:56 -05:00
funnel-transformer_tf
Remove "gpu" device alias and migrate to using "cuda" for NVIDIA GPU. (#325)
2022-09-13 01:16:56 -05:00
google_vit-base-patch16-224_tf
Remove "gpu" device alias and migrate to using "cuda" for NVIDIA GPU. (#325)
2022-09-13 01:16:56 -05:00
google_vit-base-patch16-224_torch
Remove "gpu" device alias and migrate to using "cuda" for NVIDIA GPU. (#325)
2022-09-13 01:16:56 -05: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
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05:00
longformer-base-4096_tf
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05:00
microsoft_beit-base-patch16-224-pt22k-ft22k_torch
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05:00
microsoft_resnet-50_torch
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05:00
midas
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
MiniLM-L12-H384-uncased
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05:00
MiniLM-L12-H384-uncased_torch
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -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
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05:00
mobilebert-uncased_torch
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05: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
Remove "gpu" device alias and migrate to using "cuda" for NVIDIA GPU. (#325)
2022-09-13 01:16:56 -05: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
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05: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
nvidia_mit-b0_torch
Remove "gpu" device alias and migrate to using "cuda" for NVIDIA GPU. (#325)
2022-09-13 01:16:56 -05:00
pytorch
Remove "gpu" device alias and migrate to using "cuda" for NVIDIA GPU. (#325)
2022-09-13 01:16:56 -05:00
rembert_tf
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05:00
resnet18_torch
Update xfails for torchvision models. (#310)
2022-09-01 13:06:12 -05:00
resnet50
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05:00
resnet50_torch
Update xfails for torchvision models. (#310)
2022-09-01 13:06:12 -05:00
resnet101_torch
Update xfails for torchvision models. (#310)
2022-09-01 13:06:12 -05: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
Remove "gpu" device alias and migrate to using "cuda" for NVIDIA GPU. (#325)
2022-09-13 01:16:56 -05:00
squeezenet
Remove tf package dep for SharkDownloader tflite tests (#212)
2022-07-26 12:41:12 -07:00
squeezenet1_0_torch
Update xfails for torchvision models. (#310)
2022-09-01 13:06:12 -05: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
tiny-random-flaubert_tf
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -05:00
wide_resnet50_2_torch
Update xfails for torchvision models. (#310)
2022-09-01 13:06:12 -05:00
xlm-roberta-base_tf
Remove "gpu" device alias and migrate to using "cuda" for NVIDIA GPU. (#325)
2022-09-13 01:16:56 -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 huggingface top5 image classification automodel (#268)
2022-08-22 15:05:38 -07:00
README.md
fine tune with shark (#211)
2022-08-04 13:14:57 -05:00
test_utils.py
Simplified Testing Interface (#289)
2022-08-24 23:54:56 -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