Skip to content
This repository was archived by the owner on May 12, 2024. It is now read-only.

Commit ae2f1d9

Browse files
committed
tftrt float32 bug fixes
1 parent 55cbd58 commit ae2f1d9

File tree

3 files changed

+4
-2
lines changed

3 files changed

+4
-2
lines changed

Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ ARG CPVER=cp38
77
ARG OPENVINOVER=2021.4.582
88
ARG OPENVINOROOTDIR=/opt/intel/openvino_2021
99
ARG TENSORRTVER=cuda11.3-trt8.0.1.6-ga-20210626
10-
ARG APPVER=v1.13.4
10+
ARG APPVER=v1.13.5
1111
ARG wkdir=/home/user
1212

1313
# dash -> bash

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
setup(
1212
name="tflite2tensorflow",
1313
scripts=scripts,
14-
version="1.13.4",
14+
version="1.13.5",
1515
description="Generate saved_model, tfjs, tf-trt, EdgeTPU, CoreML, quantized tflite, ONNX, OpenVINO, Myriad Inference Engine blob and .pb from .tflite.",
1616
long_description=long_description,
1717
long_description_content_type="text/markdown",

tflite2tensorflow/tflite2tensorflow.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6005,6 +6005,8 @@ def input_fn():
60056005
if input_shapes_permutations_idx > len(input_shapes_permutations):
60066006
print(f'{Color.RED}ERROR:{Color.RESET}', e)
60076007

6008+
converter.save(f'{model_output_path}/tensorrt_saved_model_float32')
6009+
print(f'{Color.GREEN}TF-TRT (TensorRT) convertion complete!{Color.RESET} - {model_output_path}/tensorrt_saved_model_float32')
60086010
except Exception as e:
60096011
print(f'{Color.RED}ERROR:{Color.RESET}', e)
60106012
import traceback

0 commit comments

Comments
 (0)