AMD-SHARK-Studio

mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00

Files

Prashant Kumar 3c2f8ef243 Generate shark_tank for tensorflow models.

Updated the generate_sharktank.py script to give tensorflow models.

2022-07-20 22:37:14 +05:30

hf_masked_lm

Update TF pytest marks and fix benchmark runner import. (#196 )

2022-07-19 15:42:32 -05:00

automodelmaskedlm.py

Add shark_importer for torch_models. (#183 )

2022-07-12 20:38:19 -07:00

bert_large_gen.py

Add shark_importer for torch_models. (#183 )

2022-07-12 20:38:19 -07:00

bert_large_run.py

Add shark_importer for torch_models. (#183 )

2022-07-12 20:38:19 -07:00

bert_large_tf.py

Add shark_importer for torch_models. (#183 )

2022-07-12 20:38:19 -07:00

bert_small_gen.py

Add shark_importer for torch_models. (#183 )

2022-07-12 20:38:19 -07:00

bert_small_run.py

Add shark_importer for torch_models. (#183 )

2022-07-12 20:38:19 -07:00

bert_small_tf_run.py

Add shark_importer for torch_models. (#183 )

2022-07-12 20:38:19 -07:00

huggingface_MiniLM_gen.py

Add shark_importer for torch_models. (#183 )

2022-07-12 20:38:19 -07:00

huggingface_MiniLM_run.py

Add shark_importer for torch_models. (#183 )

2022-07-12 20:38:19 -07:00

huggingface_MiniLM_tf.py

Add shark_importer for torch_models. (#183 )

2022-07-12 20:38:19 -07:00

minilm_tf_test.py

Add shark_importer for torch_models. (#183 )

2022-07-12 20:38:19 -07:00

README.md

Add seq_classification shark module to run seq_classification

2022-06-14 22:36:19 +05:30

seq_classification.py

Add shark_importer for torch_models. (#183 )

2022-07-12 20:38:19 -07:00

tf_model_list.csv

Generate shark_tank for tensorflow models.

2022-07-20 22:37:14 +05:30

README.md

Running SharkInference on CPUs, GPUs and MAC.

Run the binary sequence_classification.

The models supported are: hugging face sequence classification

./seq_classification.py --hf_model_name="hf_model" --device="cpu" # Use gpu | vulkan

Once the model is compiled to run on the device mentioned, we can pass in text and get the logits.