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

torch fix in gen_shark_tank (#180)

Browse Source
This commit is contained in:
Chi_Liu
2022-07-08 10:13:58 -07:00
committed by GitHub
parent fa7ee7e099
commit b98c9ee484
3 changed files with 12 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
generate_sharktank.py
View File

@@ -94,6 +94,9 @@ class SharkTank:
torch_model_name_dir = os.path.join(self.workdir, str(torch_model_name))
os.makedirs(torch_model_name_dir, exist_ok=True)
print(f"TMP_TORCH_MODELNAME_DIR = {torch_model_name_dir}")
loc = torch_model_name.find("/") + 1
torch_model_name = torch_model_name[loc:]
torch_file = "/".join(
[
torch_model_name_dir,

7
tank/pytorch/torch_model_blacklist.csv Normal file
View File

@@ -0,0 +1,7 @@
distilbert-base-uncased,False,True
models.alexnet,False,False
models.resnet18,False,False
models.resnet50,False,False
models.resnet101,False,False
models.squeezenet1_0,False,False
models.wide_resnet50_2,False,False
1 distilbert-base-uncased False True
2 models.alexnet False False
3 models.resnet18 False False
4 models.resnet50 False False
5 models.resnet101 False False
6 models.squeezenet1_0 False False
7 models.wide_resnet50_2 False False

11
tank/pytorch/torch_model_list.csv
View File

@@ -1,10 +1,3 @@
albert-base-v2,False,True
bert-base-uncased,False,True
distilbert-base-uncased,False,True
microsoft/MiniLM-L12-H384-uncased,False,True
models.alexnet,False,False
models.resnet18,False,False
models.resnet50,False,False
models.resnet101,False,False
models.squeezenet1_0,False,False
models.wide_resnet50_2,False,False
albert-base-v2,False,True
bert-base-uncased,False,True
1 albert-base-v2 microsoft/MiniLM-L12-H384-uncased False True
albert-base-v2 False True
bert-base-uncased False True
distilbert-base-uncased False True
1 microsoft/MiniLM-L12-H384-uncased microsoft/MiniLM-L12-H384-uncased False True
2 models.alexnet albert-base-v2 False False True
3 models.resnet18 bert-base-uncased False False True
models.resnet50 False False
models.resnet101 False False
models.squeezenet1_0 False False
models.wide_resnet50_2 False False