Merge pull request #926 from FlyingQianMM/develop_qh

change to eval mode when load a detection model before initializing a pruner

dygraph/paddlex/__init__.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = '2.0.0rc1'
+__version__ = '2.0.0rc3'
 
 from paddlex.utils.env import get_environ_info, init_parallel_env
 init_parallel_env()

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

dygraph/setup.py

@@ -19,7 +19,7 @@ long_description = "PaddlePaddle Entire Process Development Toolkit"
 
 setuptools.setup(
     name="paddlex",
-    version='2.0.0rc2',
+    version='2.0.0rc3',
     author="paddlex",
     author_email="paddlex@baidu.com",
     description=long_description,