Skip to content

Commit b3e4bf0

Browse files
authored
Update some PyTorch examples for new API (#191)
Signed-off-by: Cheng, Penghui <[email protected]>
1 parent 2533a45 commit b3e4bf0

File tree

410 files changed

+325
-43899
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

410 files changed

+325
-43899
lines changed

examples/.config/model_params_pytorch.json

Lines changed: 5 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -80,28 +80,15 @@
8080
"model_src_dir": "nlp/huggingface_models/question-answering/quantization/ptq_static/ipex",
8181
"dataset_location": "",
8282
"input_model": "",
83-
"yaml": "conf.yaml",
84-
"strategy": "basic",
83+
"main_script": "run_qa.py",
8584
"batch_size": 8,
86-
"new_benchmark": false
8785
},
8886
"bert_large_ipex":{
8987
"model_src_dir": "nlp/huggingface_models/question-answering/quantization/ptq_static/ipex",
9088
"dataset_location": "",
9189
"input_model": "",
92-
"yaml": "conf.yaml",
93-
"strategy": "basic",
90+
"main_script": "run_qa.py",
9491
"batch_size": 8,
95-
"new_benchmark": false
96-
},
97-
"bert_large_1_10_ipex":{
98-
"model_src_dir": "nlp/huggingface_models/question-answering/quantization/ptq_static/ipex",
99-
"dataset_location": "/tf_dataset/pytorch/squad/dev-v1.1.json",
100-
"input_model": "/tf_dataset/pytorch/bert_large_ipex",
101-
"yaml": "conf.yaml",
102-
"strategy": "basic",
103-
"batch_size": 8,
104-
"new_benchmark": false
10592
},
10693
"resnet50_gpu": {
10794
"model_src_dir": "image_recognition/torchvision_models/quantization/ptq/gpu/eager/",
@@ -285,10 +272,8 @@
285272
"model_src_dir": "nlp/huggingface_models/question-answering/quantization/ptq_static/fx",
286273
"dataset_location": "",
287274
"input_model": "",
288-
"yaml": "conf.yaml",
289-
"strategy": "basic",
275+
"main_script": "run_qa.py",
290276
"batch_size": 64,
291-
"new_benchmark": false
292277
},
293278
"bert_large_QNLI": {
294279
"model_src_dir": "nlp/huggingface_models/text-classification/quantization/ptq_static/fx",
@@ -546,10 +531,8 @@
546531
"model_src_dir": "nlp/huggingface_models/language-modeling/quantization/ptq_static/fx",
547532
"dataset_location": "",
548533
"input_model": "/tf_dataset2/models/pytorch/gpt-j-6B",
549-
"yaml": "conf.yaml",
550-
"strategy": "basic",
551-
"batch_size": 8,
552-
"new_benchmark": false
534+
"main_script": "run_clm.py",
535+
"batch_size": 8
553536
},
554537
"xlm-roberta-base_MRPC": {
555538
"model_src_dir": "nlp/huggingface_models/text-classification/quantization/ptq_static/eager",

examples/README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -445,7 +445,7 @@ Intel® Neural Compressor validated examples with multiple compression technique
445445
<td>3D-UNet</td>
446446
<td>Image Recognition</td>
447447
<td>Post-Training Static Quantization</td>
448-
<td><a href="./pytorch/image_recognition/3d-unet/quantization/ptq/eager">eager</a></td>
448+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/pytorch/image_recognition/3d-unet/quantization/ptq/eager">eager</a></td>
449449
</tr>
450450
<tr>
451451
<td>SSD ResNet34</td>
@@ -463,7 +463,7 @@ Intel® Neural Compressor validated examples with multiple compression technique
463463
<td>YOLOv3</td>
464464
<td>Object Detection</td>
465465
<td>Post-Training Static Quantization</td>
466-
<td><a href="./pytorch/object_detection/yolo_v3/quantization/ptq/eager">eager</a></td>
466+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/pytorch/object_detection/yolo_v3/quantization/ptq/eager">eager</a></td>
467467
</tr>
468468
<tr>
469469
<td>DLRM</td>
@@ -493,7 +493,7 @@ Intel® Neural Compressor validated examples with multiple compression technique
493493
<td>BlendCNN</td>
494494
<td>Natural Language Processing</td>
495495
<td>Post-Training Static Quantization</td>
496-
<td><a href="./pytorch/nlp/blendcnn/quantization/ptq/eager">eager</a></td>
496+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/pytorch/nlp/blendcnn/quantization/ptq/eager">eager</a></td>
497497
</tr>
498498
<tr>
499499
<td>bert-large-uncased-whole-word-masking-finetuned-squad</td>

examples/pytorch/image_recognition/3d-unet/quantization/ptq/eager/.dockerignore

Lines changed: 0 additions & 1 deletion
This file was deleted.

examples/pytorch/image_recognition/3d-unet/quantization/ptq/eager/.gitignore

Lines changed: 0 additions & 1 deletion
This file was deleted.

examples/pytorch/image_recognition/3d-unet/quantization/ptq/eager/3DUnet_README.md

Lines changed: 0 additions & 52 deletions
This file was deleted.

examples/pytorch/image_recognition/3d-unet/quantization/ptq/eager/Makefile

Lines changed: 0 additions & 204 deletions
This file was deleted.

0 commit comments

Comments
 (0)