Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 0 additions & 1 deletion .azure-pipelines/scripts/fwk_version.sh
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
#!/bin/bash

echo "export FWs version..."
export tensorflow_version='2.15.0-official'
export pytorch_version='2.5.1'
export torchvision_version='0.20.1'
export ipex_version='2.5.0+cpu'
Expand Down
5 changes: 0 additions & 5 deletions .azure-pipelines/scripts/install_nc.sh
Original file line number Diff line number Diff line change
Expand Up @@ -23,11 +23,6 @@ if [[ $1 = *"3x_pt"* ]]; then
fi
python setup.py pt bdist_wheel
uv pip install --no-deps dist/neural_compressor*.whl --force-reinstall
elif [[ $1 = *"3x_tf"* ]]; then
uv pip install tensorflow==2.19.0
uv pip install --no-cache-dir -r requirements_tf.txt
python setup.py tf bdist_wheel
uv pip install dist/neural_compressor*.whl --force-reinstall
elif [[ $1 = *"3x_jax"* ]]; then
uv pip install --no-cache-dir -r requirements_jax.txt
python setup.py jax bdist_wheel
Expand Down
5 changes: 0 additions & 5 deletions .azure-pipelines/scripts/models/collect_log_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,6 @@
+ args.build_id
+ "&view=artifacts&pathAsName=false&type=publishedArtifacts"
)
OOB_MODEL_LIST = ["darknet19", "densenet-121", "resnet-101"]


def get_model_tuning_dict_results():
Expand Down Expand Up @@ -141,10 +140,6 @@ def collect_log():
parse_tuning_line(line, tmp)
print(tmp)

# oob_model no need acc
if (args.model in OOB_MODEL_LIST) and args.framework == "tensorflow":
tmp["fp32_acc"], tmp["int8_acc"] = "unknown", "unknown"

# set for 3x woq models
if args.inc_new_api.split("_")[0] == "3x":
tmp["fp32_acc"], tmp["tuning_trials"], tmp["strategy"] = "unknown", "", ""
Expand Down
2 changes: 1 addition & 1 deletion .azure-pipelines/scripts/models/generate_report.sh
Original file line number Diff line number Diff line change
Expand Up @@ -530,7 +530,7 @@ cat > ${output_dir}/report.html << eof
<html lang="en">
<head>
<meta http-equiv="content-type" content="text/html; charset=ISO-8859-1">
<title>Daily Tests - TensorFlow - Jenkins</title>
<title>Daily Tests - Neural Compressor - Jenkins</title>
<style type="text/css">
body
{
Expand Down
4 changes: 0 additions & 4 deletions .azure-pipelines/scripts/models/run_model_trigger_common.sh
Original file line number Diff line number Diff line change
Expand Up @@ -80,8 +80,6 @@ if [ "${mode}" == "env_setup" ]; then
elif [ "${mode}" == "tuning" ]; then
if [ "${framework}" == "onnxrt" ]; then
output_model=${log_dir}/${model}/${framework}-${model}-tune.onnx
elif [ "${framework}" == "tensorflow" ]; then
output_model=${log_dir}/${model}/${framework}-${model}-tune.pb
fi
[[ ${output_model} ]] && tuning_cmd="${tuning_cmd} --output_model=${output_model}"

Expand Down Expand Up @@ -138,8 +136,6 @@ elif [ "${mode}" == "int8_benchmark" ]; then
$BOLD_YELLOW && echo "====== run benchmark int8 =======" && $RESET
if [[ "${framework}" == "onnxrt" ]]; then
model_name="${log_dir}/${model}/${framework}-${model}-tune.onnx"
elif [[ "${framework}" == "tensorflow" ]]; then
model_name="${log_dir}/${model}/${framework}-${model}-tune.pb"
elif [[ "${framework}" == "pytorch" ]]; then
model_name=${input_model}
benchmark_cmd="${benchmark_cmd} --int8=true"
Expand Down
10 changes: 0 additions & 10 deletions .azure-pipelines/scripts/models/update_yaml_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,7 +100,6 @@ def update_yaml_config_tuning(
strategy_token=None,
sampling_size=None,
dtype=None,
tf_new_api=None,
):
with open(yaml_file) as f:
yaml_config = yaml.round_trip_load(f, preserve_quotes=True)
Expand Down Expand Up @@ -216,15 +215,6 @@ def update_yaml_config_tuning(
except Exception as e:
print(f"[ WARNING ] {e}")

if tf_new_api == "true":
try:
model = yaml_config.get("model", {})
prev_framework = model.get("framework", None)
model.update({"framework": "inteltensorflow"})
print(f"Changed framework from {prev_framework} to inteltensorflow")
except Exception as e:
print(f"[ WARNING ] {e}")

print("====== update_yaml_config_tuning ========")

yaml_content = yaml.round_trip_dump(yaml_config)
Expand Down
21 changes: 0 additions & 21 deletions .azure-pipelines/scripts/ut/coverage.3x_tf

This file was deleted.

78 changes: 0 additions & 78 deletions .azure-pipelines/scripts/ut/run_3x_tf.sh

This file was deleted.

2 changes: 1 addition & 1 deletion .azure-pipelines/template/model-template.yml
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ parameters:
default: "resnet50v1.5"
- name: framework
type: string
default: "tensorflow"
default: "pytorch"
- name: APIVersion
type: string
default: ""
Expand Down
118 changes: 0 additions & 118 deletions .azure-pipelines/ut-3x-tf.yml

This file was deleted.

16 changes: 1 addition & 15 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

Intel® Neural Compressor
===========================
<h3> An open-source Python library supporting popular model compression techniques on mainstream deep learning frameworks (PyTorch, TensorFlow, and JAX)</h3>
<h3> An open-source Python library supporting popular model compression techniques on mainstream deep learning frameworks (PyTorch and JAX)</h3>

[![python](https://img.shields.io/badge/python-3.11%2B-blue)](https://github.com/intel/neural-compressor)
[![version](https://img.shields.io/badge/release-3.10-green)](https://github.com/intel/neural-compressor/releases)
Expand Down Expand Up @@ -47,8 +47,6 @@ Intel Neural Compressor supports PyTorch with CPU, GPU and HPU. Please install t
```
# Framework extension API + PyTorch dependency
pip install neural-compressor-pt
# Framework extension API + TensorFlow dependency
pip install neural-compressor-tf
# Framework extension API + JAX dependency
pip install neural-compressor-jax
```
Expand Down Expand Up @@ -144,18 +142,6 @@ model = load(
<td colspan="4" align="center"><a href="./docs/source/PT_NVFP4Quant.md">NVFP4 Quantization</a></td>
</tr>
</tbody>
<thead>
<tr>
<th colspan="8">Tensorflow Extension APIs</th>
</tr>
</thead>
<tbody>
<tr>
<td colspan="3" align="center"><a href="./docs/source/TensorFlow.md">Overview</a></td>
<td colspan="3" align="center"><a href="./docs/source/TF_Quant.md">Static Quantization</a></td>
<td colspan="2" align="center"><a href="./docs/source/TF_SQ.md">Smooth Quantization</a></td>
</tr>
</tbody>
<thead>
<tr>
<th colspan="8">Transformers-like APIs</th>
Expand Down
Loading
Loading