Explorar o código

change eval() when load a detector model

FlyingQianMM %!s(int64=4) %!d(string=hai) anos
pai
achega
65ccb588a6
Modificáronse 1 ficheiros con 1 adicións e 0 borrados
  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']