Skip to content

Commit 747068b

Browse files
authored
Merge pull request #9 from sharvil10/sharvils/tf_notebook_fix
TF notebook fix
2 parents 61254cb + a53947c commit 747068b

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

‎Intel_Optimized_TensorFlow_and_INC_Quantization_Tool_E2E_Sample/inc_sample_tensorflow.ipynb‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -185,7 +185,7 @@
185185
"source": [
186186
"from tensorflow.python.framework.convert_to_constants import convert_variables_to_constants_v2\n",
187187
"\n",
188-
"def save_frezon_pb(model, mod_path):\n",
188+
"def save_frozen_pb(model, mod_path):\n",
189189
" # Convert Keras model to ConcreteFunction\n",
190190
" full_model = tf.function(lambda x: model(x))\n",
191191
" concrete_function = full_model.get_concrete_function(\n",
@@ -199,8 +199,8 @@
199199
" logdir=\".\",\n",
200200
" name=mod_path,\n",
201201
" as_text=False)\n",
202-
"fp32_frezon_pb_file = \"fp32_frozen.pb\"\n",
203-
"save_frezon_pb(model, fp32_frezon_pb_file)"
202+
"fp32_frozen_pb_file = \"fp32_frozen.pb\"\n",
203+
"save_frozen_pb(model, fp32_frozen_pb_file)"
204204
]
205205
},
206206
{
@@ -209,7 +209,7 @@
209209
"metadata": {},
210210
"outputs": [],
211211
"source": [
212-
"!ls -la fp32_frezon.pb"
212+
"!ls -la fp32_frozen.pb"
213213
]
214214
},
215215
{
@@ -328,7 +328,7 @@
328328
" return q_model\n",
329329
"\n",
330330
"\n",
331-
"def save_int8_frezon_pb(q_model, path):\n",
331+
"def save_int8_frozen_pb(q_model, path):\n",
332332
" from tensorflow.python.platform import gfile\n",
333333
" f = gfile.GFile(path, 'wb')\n",
334334
" f.write(q_model.as_graph_def().SerializeToString())\n",
@@ -421,7 +421,7 @@
421421
},
422422
"outputs": [],
423423
"source": [
424-
"!python profiling_inc.py --input-graph=./fp32_frezon.pb --omp-num-threads=4 --num-inter-threads=1 --num-intra-threads=4 --index=32"
424+
"!python profiling_inc.py --input-graph=./fp32_frozen.pb --omp-num-threads=4 --num-inter-threads=1 --num-intra-threads=4 --index=32"
425425
]
426426
},
427427
{

0 commit comments

Comments
 (0)