From 9d83550b670df49395bc8ba863a736b01a836d6a Mon Sep 17 00:00:00 2001
From: liaoxingyu <sherlockliao01@gmail.com>
Date: Fri, 26 Mar 2021 10:10:45 +0800
Subject: [PATCH] bugfix

Summary: make `deploy` adaptation for other models except repvgg
---
 tools/deploy/onnx_export.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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)