Fix decouple_seg export bug.

nepeplwu · nepeplwu · commit 34c1bbf545cb · 2021-06-23T10:55:46.000+08:00
diff --git a/paddleseg/models/decoupled_segnet.py b/paddleseg/models/decoupled_segnet.py
@@ -79,7 +79,9 @@ def forward(self, x):
                 align_corners=self.align_corners) for logit in logit_list
         ]
 
-        return [seg_logit, body_logit, edge_logit, (seg_logit, edge_logit)]
+        if self.training:
+            return [seg_logit, body_logit, edge_logit, (seg_logit, edge_logit)]
+        return [seg_logit]
 
     def init_weight(self):
         if self.pretrained is not None:

Original file line number	Diff line number	Diff line change
`@@ -79,7 +79,9 @@ def forward(self, x):`
`79`	`79`	`align_corners=self.align_corners) for logit in logit_list`
`80`	`80`	`]`
`81`	`81`
`82`		`- return [seg_logit, body_logit, edge_logit, (seg_logit, edge_logit)]`
	`82`	`+ if self.training:`
	`83`	`+ return [seg_logit, body_logit, edge_logit, (seg_logit, edge_logit)]`
	`84`	`+ return [seg_logit]`
`83`	`85`
`84`	`86`	`def init_weight(self):`
`85`	`87`	`if self.pretrained is not None:`