diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index dcb97c60d..679b7140c 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -299,7 +299,7 @@ class Exporter: y = model(im) # dry runs if self.args.half and onnx and self.device.type != "cpu": im, model = im.half(), model.half() # to FP16 - + if mct and "C2f" not in model.__str__(): raise ValueError("MCT export is only supported for YOLOv8 detection models") diff --git a/ultralytics/utils/benchmarks.py b/ultralytics/utils/benchmarks.py index 753705699..887b65b0a 100644 --- a/ultralytics/utils/benchmarks.py +++ b/ultralytics/utils/benchmarks.py @@ -91,9 +91,9 @@ def benchmark( t0 = time.time() for i, (name, format, suffix, cpu, gpu) in enumerate(zip(*export_formats().values())): emoji, filename = "❌", None # export defaults - print("%"*50) + print("%" * 50) print(i, name, format, suffix, cpu, gpu) - print("%"*50) + print("%" * 50) try: # Checks if i == 7: # TF GraphDef