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