diff --git a/tools/deploy/onnx_export.py b/tools/deploy/onnx_export.py index 038d7d5..75f5e8f 100644 --- a/tools/deploy/onnx_export.py +++ b/tools/deploy/onnx_export.py @@ -144,7 +144,8 @@ if __name__ == '__main__': cfg.MODEL.HEADS.POOL_LAYER = 'avgpool' model = build_model(cfg) Checkpointer(model).load(cfg.MODEL.WEIGHTS) - model.backbone.deploy(True) + if hasattr(model.backbone, 'deploy'): + model.backbone.deploy(True) model.eval() logger.info(model)