Browse Source

change eval() when load a detector model

FlyingQianMM 4 years ago
parent
commit
65ccb588a6
1 changed files with 1 additions and 0 deletions
  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']