Fix --mean_values and --scale_values prohibition to align with C++ (#20910)

This commit is contained in:
Zlobin Vladimir authored and GitHub committed 2023-11-13 16:15:02 +04:00
1 parent 03e48f3c17
commit 0125c89738
1 file changed
+2 -2
@@ -328,7 +328,7 @@ def main():
topology_name = ""
load_from_file_enabled = is_flag_set_in_command_line('load_from_file') or is_flag_set_in_command_line('lfile')
if load_from_file_enabled and not is_network_compiled:
if not args.mean_values or not args.scale_values:
if args.mean_values or args.scale_values:
raise RuntimeError("--mean_values and --scale_values aren't supported with --load_from_file. "
"The values can be set via model_optimizer while generating xml")
next_step()
@@ -414,7 +414,7 @@ def main():
('compile model time (ms)', duration_ms)
])
else:
if not args.mean_values or not args.scale_values:
if args.mean_values or args.scale_values:
raise RuntimeError("--mean_values and --scale_values aren't supported for compiled model. "
"The values can be set via model_optimizer while generating xml")
next_step()