From 7b36e38cf8f3d3c08e973b18913ae8e41ff970b2 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 20 May 2021 15:22:36 +0200 Subject: [PATCH] Check CoreML models.train() mode (#3262) * Check CoreML models.train() mode * Update export.py --- models/export.py | 1 + 1 file changed, 1 insertion(+) diff --git a/models/export.py b/models/export.py index de4adb6a7e24..65721f65d888 100644 --- a/models/export.py +++ b/models/export.py @@ -131,6 +131,7 @@ import coremltools as ct print(f'{prefix} starting export with coremltools {ct.__version__}...') + assert opt.train, 'CoreML exports should be placed in model.train() mode with `python export.py --train`' model = ct.convert(ts, inputs=[ct.ImageType('image', shape=img.shape, scale=1 / 255.0, bias=[0, 0, 0])]) f = opt.weights.replace('.pt', '.mlmodel') # filename model.save(f)