UltralyticsAssistant 2 months ago
parent 945dfd1996
commit 9edbcb7dec
  1. 2
      ultralytics/engine/exporter.py
  2. 4
      ultralytics/utils/benchmarks.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")

@ -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

Loading…
Cancel
Save