update rep condition

2022-06-28 05:58:14 +00:00 · 2022-06-28 05:58:14 +00:00 · e8fc16d76b
parent de1e9cab74
commit e8fc16d76b
1 changed files with 3 additions and 4 deletions
--- a/ppcls/engine/engine.py
+++ b/ppcls/engine/engine.py
@ -465,10 +465,9 @@ class Engine(object):
        model.eval()

        # for rep nets
-        if "Slim" not in self.config or self.config["Slim"]["quant"]["name"] != "pact":
-            for layer in self.model.sublayers():
-                if hasattr(layer, "rep"):
-                    layer.rep()
+        for layer in self.model.sublayers():
+            if hasattr(layer, "rep") and not getattr(layer, "is_repped"):
+                layer.rep()

        save_path = os.path.join(self.config["Global"]["save_inference_dir"],
                                 "inference")