added mct dependency

mct-2.1.1
Francesco Mattioli 3 months ago
parent 8a3d1b1987
commit 60c66f24db
  1. 1
      ultralytics/engine/exporter.py

@ -1035,6 +1035,7 @@ class Exporter:
@try_export @try_export
def export_mct(self, prefix=colorstr("Sony MCT:")): def export_mct(self, prefix=colorstr("Sony MCT:")):
# pip install --upgrade -force-reinstall git+https://github.com/ambitious-octopus/model_optimization.git@get-output-fix # pip install --upgrade -force-reinstall git+https://github.com/ambitious-octopus/model_optimization.git@get-output-fix
check_requirements("mct-nightly")
import model_compression_toolkit as mct import model_compression_toolkit as mct
import onnx import onnx
from model_compression_toolkit.core.pytorch.pytorch_device_config import get_working_device, set_working_device from model_compression_toolkit.core.pytorch.pytorch_device_config import get_working_device, set_working_device

Loading…
Cancel
Save