bugfix

Summary: make `deploy` adaptation for other models except repvgg
2021-03-26 10:10:45 +08:00 · 2021-03-26 10:10:45 +08:00 · 9d83550b67
parent 15c556c43a
commit 9d83550b67
1 changed files with 2 additions and 1 deletions
--- 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)