소스 검색

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']