Skip to content

Commit

Permalink
Temporarily disable new FP16 generation
Browse files Browse the repository at this point in the history
  • Loading branch information
mvafin committed Nov 17, 2021
1 parent 9c6cf61 commit bee2503
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
3 changes: 2 additions & 1 deletion model-optimizer/mo/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -205,7 +205,8 @@ def raise_ie_not_found():
except Exception as e:
raise_ie_not_found()

if 'data_type' in argv and argv.data_type in ['FP16', 'half']:
# temporary disable new FP16 generation
if False and 'data_type' in argv and argv.data_type in ['FP16', 'half']:
argv.data_type = 'FP32'
argv.compress_fp16 = True
else:
Expand Down
3 changes: 2 additions & 1 deletion model-optimizer/mo/pipeline/common.py
Original file line number Diff line number Diff line change
Expand Up @@ -180,7 +180,8 @@ def prepare_emit_ir(graph: Graph, data_type: str, output_dir: str, output_model_
meta_info = {}
graph.strict_mode = False

if not used_by_ir_reader:
# temporary disable new FP16 generation
if True and not used_by_ir_reader:
# convert Parameter data types
convert_data_type.convert_parameters_data_type(graph, data_type)
# convert blobs (usually weights and biases)
Expand Down

0 comments on commit bee2503

Please sign in to comment.