Skip to content

Update Examples for TF 3x API #1901

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 16 commits into from
Jul 25, 2024
Merged

Update Examples for TF 3x API #1901

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
f23de0d
Update Examples for TF 3x API
zehao-intel Jul 4, 2024
00ef387
add keras models
zehao-intel Jul 9, 2024
e82746f
fix import issues
zehao-intel Jul 11, 2024
8c90766
fix config
zehao-intel Jul 11, 2024
60fc76c
fix keras config
zehao-intel Jul 11, 2024
b78e20b
fix multiple issues
zehao-intel Jul 12, 2024
2406dff
Merge branch 'master' of https://github.com/intel/neural-compressor
zehao-intel Jul 15, 2024
7d3fa73
Merge branch 'master' into zehao/3x_example
zehao-intel Jul 15, 2024
88d8600
fix transformer_lt
zehao-intel Jul 15, 2024
b92543c
fix keras
zehao-intel Jul 15, 2024
6c5ae4c
remove distillbert
zehao-intel Jul 16, 2024
5f50a34
add save log for keras
zehao-intel Jul 16, 2024
cb40371
fix accuracy using per-channel
zehao-intel Jul 18, 2024
5307820
fix keras accuracy
zehao-intel Jul 23, 2024
150b7b4
remove densenet
zehao-intel Jul 24, 2024
f37f341
change folder name of examples
zehao-intel Jul 25, 2024
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
Prev Previous commit
Next Next commit
add save log for keras 
Signed-off-by: zehao-intel <[email protected]>
  • Loading branch information
@zehao-intel
zehao-intel committed Jul 16, 2024
commit 5f50a34b18eefb67d5a43f529f1000201b212168
2 changes: 1 addition & 1 deletion examples/3.x_api/tensorflow/keras/cv/inception_v3/quantization/ptq/main.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -126,7 +126,7 @@ def main(_):
quant_config = StaticQuantConfig()
q_model = quantize_model(FLAGS.input_model, quant_config, calib_dataloader)
q_model.save(FLAGS.output_model)

logger.info("Save quantized model to {}.".format(FLAGS.output_model))

if FLAGS.benchmark:
from neural_compressor.tensorflow import Model
Expand Down
1 change: 1 addition & 0 deletions examples/3.x_api/tensorflow/keras/cv/mobilenet_v2/quantization/ptq/main.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -126,6 +126,7 @@ def main(_):
quant_config = StaticQuantConfig()
q_model = quantize_model(FLAGS.input_model, quant_config, calib_dataloader)
q_model.save(FLAGS.output_model)
logger.info("Save quantized model to {}.".format(FLAGS.output_model))

if FLAGS.benchmark:
from neural_compressor.tensorflow import Model
Expand Down
1 change: 1 addition & 0 deletions examples/3.x_api/tensorflow/keras/cv/resnet_v2_50/quantization/ptq/main.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -126,6 +126,7 @@ def main(_):
quant_config = StaticQuantConfig()
q_model = quantize_model(FLAGS.input_model, quant_config, calib_dataloader)
q_model.save(FLAGS.output_model)
logger.info("Save quantized model to {}.".format(FLAGS.output_model))

if FLAGS.benchmark:
from neural_compressor.tensorflow import Model
Expand Down