Pārlūkot izejas kodu

change eval() when load a detector model

FlyingQianMM 4 gadi atpakaļ
vecāks
revīzija
65ccb588a6
1 mainītis faili ar 1 papildinājumiem un 0 dzēšanām
  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']