From 7a96b6d1bcb1ad4b98b11a200046ecf769149737 Mon Sep 17 00:00:00 2001
From: littletomatodonkey <2120160898@bit.edu.cn>
Date: Thu, 6 May 2021 13:55:37 +0800
Subject: [PATCH] adapt to net.eval for the framework just contains training
 flag setting (#711)

---
 tools/export_model.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/tools/export_model.py b/tools/export_model.py
index 51b4fe2b..5d6b338d 100644
--- a/tools/export_model.py
+++ b/tools/export_model.py
@@ -47,6 +47,12 @@ class Net(paddle.nn.Layer):
         self.pre_net = net(class_dim=class_dim)
         self.model = model
 
+    def eval(self):
+        self.training = False
+        for layer in self.sublayers():
+            layer.training = False
+            layer.eval()
+
     def forward(self, inputs):
         x = self.pre_net(inputs)
         if self.model == "GoogLeNet":
-- 
GitLab