|
|
@ -201,7 +201,7 @@ class Exporter:
|
|
|
|
if self.args.half and (engine or onnx) and self.device.type != 'cpu':
|
|
|
|
if self.args.half and (engine or onnx) and self.device.type != 'cpu':
|
|
|
|
im, model = im.half(), model.half() # to FP16
|
|
|
|
im, model = im.half(), model.half() # to FP16
|
|
|
|
|
|
|
|
|
|
|
|
# Warnings
|
|
|
|
# Filter warnings
|
|
|
|
warnings.filterwarnings('ignore', category=torch.jit.TracerWarning) # suppress TracerWarning
|
|
|
|
warnings.filterwarnings('ignore', category=torch.jit.TracerWarning) # suppress TracerWarning
|
|
|
|
warnings.filterwarnings('ignore', category=UserWarning) # suppress shape prim::Constant missing ONNX warning
|
|
|
|
warnings.filterwarnings('ignore', category=UserWarning) # suppress shape prim::Constant missing ONNX warning
|
|
|
|
warnings.filterwarnings('ignore', category=DeprecationWarning) # suppress CoreML np.bool deprecation warning
|
|
|
|
warnings.filterwarnings('ignore', category=DeprecationWarning) # suppress CoreML np.bool deprecation warning
|
|
|
|