fix _init_amp when export

2d8346cd · gaotingquan · Tingquan Gao · 5bb2e8c3 · 2d8346cd
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

ppcls/engine/engine.py ppcls/engine/engine.py +3 -0

未找到文件。
--- a/ppcls/engine/engine.py
+++ b/ppcls/engine/engine.py
@@ -524,6 +524,9 @@ class Engine(object):
        )

    def _init_amp(self):
+        if self.mode == "export":
+            return
+
        amp_config = self.config.get("AMP", None)
        use_amp = True if amp_config and amp_config.get("use_amp",
                                                        True) else False