浏览代码

change eval() when load a detector model

FlyingQianMM 4 年之前
父节点
当前提交
65ccb588a6
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1 0
      dygraph/paddlex/cv/models/load_model.py

+ 1 - 0
dygraph/paddlex/cv/models/load_model.py

@@ -97,6 +97,7 @@ def load_model(model_dir):
                         k: paddle.to_tensor(v)
                         for k, v in inputs.items()
                     }]
+                    model.net.eval()
                 model.pruner = getattr(paddleslim, pruning_info['pruner'])(
                     model.net, inputs=inputs)
                 model.pruning_ratios = pruning_info['pruning_ratios']