Skip to content

Commit 2af6e5c

Browse files
committed
Update examples test config file
Signed-off-by: Cheng, Penghui <[email protected]>
1 parent b34947b commit 2af6e5c

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

examples/.config/model_params_pytorch.json

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -367,7 +367,10 @@
367367
"model_src_dir": "nlp/blendcnn/quantization/ptq/eager",
368368
"dataset_location": "/tf_dataset/pytorch/blendcnn/MRPC/",
369369
"input_model": "/tf_dataset/pytorch/blendcnn/models/",
370+
"yaml": "blendcnn.yaml",
371+
"strategy": "basic",
370372
"batch_size": 32,
373+
"new_benchmark": false
371374
},
372375
"gpt_WikiText(rm)": {
373376
"model_src_dir": "eager/language_translation/ptq",
@@ -418,7 +421,10 @@
418421
"model_src_dir": "image_recognition/3d-unet/quantization/ptq/eager",
419422
"dataset_location": "/tf_dataset/dataset/mlperf_3dunet/build",
420423
"input_model": "/tf_dataset/pytorch/mlperf_3dunet/nnUNetTrainerV2__nnUNetPlansv2.mlperf.1",
421-
"batch_size": 100
424+
"yaml": "conf.yaml",
425+
"strategy": "basic",
426+
"batch_size": 100,
427+
"new_benchmark": false
422428
},
423429
"rnnt": {
424430
"model_src_dir": "speech_recognition/rnnt/quantization/ptq_dynamic/eager",

0 commit comments

Comments
 (0)