diff --git a/examples/recipes/daekeun-ml_koelectra-small-v3-nsmc/cpu/cpu/text-classification_fp16_config.json b/examples/recipes/daekeun-ml_koelectra-small-v3-nsmc/cpu/cpu/text-classification_fp16_config.json new file mode 100644 index 000000000..1904a6746 --- /dev/null +++ b/examples/recipes/daekeun-ml_koelectra-small-v3-nsmc/cpu/cpu/text-classification_fp16_config.json @@ -0,0 +1,101 @@ +{ + "export": { + "opset_version": 17, + "batch_size": 1, + "export_params": true, + "do_constant_folding": true, + "verbose": false, + "dynamo": false, + "enable_hierarchy_tags": true, + "clean_onnx": false, + "hierarchy_tag_format": "full", + "input_tensors": [ + { + "name": "input_ids", + "dtype": "int32", + "shape": [ + 1, + 512 + ], + "value_range": [ + 0, + 35000 + ] + }, + { + "name": "attention_mask", + "dtype": "int32", + "shape": [ + 1, + 512 + ], + "value_range": [ + 0, + 2 + ] + }, + { + "name": "token_type_ids", + "dtype": "int32", + "shape": [ + 1, + 512 + ], + "value_range": [ + 0, + 2 + ] + } + ], + "output_tensors": [ + { + "name": "logits" + } + ] + }, + "optim": {}, + "quant": { + "mode": "fp16", + "samples": 10, + "calibration_method": "minmax", + "weight_type": "uint8", + "activation_type": "uint8", + "per_channel": false, + "symmetric": false, + "weight_symmetric": null, + "activation_symmetric": null, + "save_calibration": false, + "distribution": "uniform", + "seed": null, + "calibration_load_path": null, + "calibration_save_path": null, + "op_types_to_quantize": null, + "nodes_to_exclude": null, + "fp16_keep_io_types": true, + "fp16_op_block_list": null + }, + "compile": null, + "loader": { + "task": "text-classification", + "model_class": "AutoModelForSequenceClassification", + "model_type": "electra" + }, + "eval": { + "task": "text-classification", + "dataset": { + "path": "e9t/nsmc", + "split": "test", + "samples": 100, + "shuffle": false, + "streaming": true, + "revision": "153263700c285ac0997b3e2a2d80b826aaf672d6", + "columns_mapping": { + "input_column": "document" + }, + "label_mapping": { + "negative": 0, + "positive": 1 + } + } + } +} \ No newline at end of file diff --git a/examples/recipes/daekeun-ml_koelectra-small-v3-nsmc/cpu/cpu/text-classification_fp32_config.json b/examples/recipes/daekeun-ml_koelectra-small-v3-nsmc/cpu/cpu/text-classification_fp32_config.json new file mode 100644 index 000000000..bcb1f07e3 --- /dev/null +++ b/examples/recipes/daekeun-ml_koelectra-small-v3-nsmc/cpu/cpu/text-classification_fp32_config.json @@ -0,0 +1,82 @@ +{ + "export": { + "opset_version": 17, + "batch_size": 1, + "export_params": true, + "do_constant_folding": true, + "verbose": false, + "dynamo": false, + "enable_hierarchy_tags": true, + "clean_onnx": false, + "hierarchy_tag_format": "full", + "input_tensors": [ + { + "name": "input_ids", + "dtype": "int32", + "shape": [ + 1, + 512 + ], + "value_range": [ + 0, + 35000 + ] + }, + { + "name": "attention_mask", + "dtype": "int32", + "shape": [ + 1, + 512 + ], + "value_range": [ + 0, + 2 + ] + }, + { + "name": "token_type_ids", + "dtype": "int32", + "shape": [ + 1, + 512 + ], + "value_range": [ + 0, + 2 + ] + } + ], + "output_tensors": [ + { + "name": "logits" + } + ] + }, + "optim": {}, + "quant": null, + "compile": null, + "loader": { + "task": "text-classification", + "model_class": "AutoModelForSequenceClassification", + "model_type": "electra" + }, + "eval": { + "task": "text-classification", + "dataset": { + "path": "e9t/nsmc", + "split": "test", + "samples": 100, + "shuffle": false, + "streaming": true, + "revision": "153263700c285ac0997b3e2a2d80b826aaf672d6", + "columns_mapping": { + "input_column": "document" + }, + "label_mapping": { + "negative": 0, + "positive": 1 + } + } + } +} \ No newline at end of file