Skip to content

Commit

Permalink
Update cle_bc.ipynb (#2590)
Browse files Browse the repository at this point in the history
In place of BN_folded_sess added cle_applied_sess, due to which segmentation fault error was resolved while doing compute encoding. By passing cle_applied_sess in Quantsim function we can run the code successfully.

Signed-off-by: Adarsh Mamidpelliwar <[email protected]>
  • Loading branch information
adarshm93 authored Jan 11, 2024
1 parent 6170393 commit 2444ab5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion Examples/tensorflow/quantization/cle_bc.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -521,7 +521,7 @@
"metadata": {},
"outputs": [],
"source": [
"sim = QuantizationSimModel(session=BN_folded_sess,\n",
"sim = QuantizationSimModel(session=cle_applied_sess,\n",
" starting_op_names=starting_op_names,\n",
" output_op_names=output_op_names,\n",
" quant_scheme= QuantScheme.training_range_learning_with_tf_enhanced_init,\n",
Expand Down

0 comments on commit 2444ab5

Please sign in to comment.