[IE][TOOLS] Fix for CCT two IR mode (#3084)
authorRoman Vyunov (Intel) <roman.vyunov@intel.com>
Tue, 17 Nov 2020 08:41:48 +0000 (11:41 +0300)
committerGitHub <noreply@github.com>
Tue, 17 Nov 2020 08:41:48 +0000 (11:41 +0300)
inference-engine/tools/cross_check_tool/cross_check_tool.py

index 4a68c3d..0bbc3ec 100644 (file)
@@ -207,8 +207,9 @@ def two_ir_mode(args):
     global_accuracy = []
     global_times, ref_global_times = overall_accuracy_check(model=args.model, ref_model=args.reference_model,
                                                             out_layers=out_layers, ref_out_layers=ref_out_layers,
-                                                            inputs=inputs, ref_inputs=ref_inputs, plugin=core,
-                                                            ref_plugin=ref_core, layers=args.layers,
+                                                            inputs=inputs, ref_inputs=ref_inputs, core=core,
+                                                            device=args.device, ref_core=ref_core,
+                                                            ref_device=args.reference_device, layers=args.layers,
                                                             num_of_iterations=args.num_of_iterations)
     for out_layer in layers_map:
         ref_out_layer = layers_map[out_layer]