From 0778b4664ad6bd4a14f31dbebd7c37df7da9db82 Mon Sep 17 00:00:00 2001
From: Guanghua Yu <742925032@qq.com>
Date: Wed, 3 Nov 2021 12:25:44 +0800
Subject: [PATCH] fix PicoDet train bug (#4439)

---
 configs/picodet/README.md               | 9 +++++++++
 ppdet/modeling/architectures/picodet.py | 2 +-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/configs/picodet/README.md b/configs/picodet/README.md
index cd25cfe64..0b7b3901a 100644
--- a/configs/picodet/README.md
+++ b/configs/picodet/README.md
@@ -195,6 +195,15 @@ python tools/post_quant.py -c configs/picodet/picodet_s_320_coco.yml \
 
 </details>
 
+## FAQ
+
+<details>
+<summary>Out of memory error.</summary>
+
+Please reduce the `batch_size` of `TrainReader` in config.
+
+</details>
+
 ## Cite PP-PiocDet
 If you use PiocDet in your research, please cite our work by using the following BibTeX entry:
 ```
diff --git a/ppdet/modeling/architectures/picodet.py b/ppdet/modeling/architectures/picodet.py
index 9c67005b0..cd807a9fa 100644
--- a/ppdet/modeling/architectures/picodet.py
+++ b/ppdet/modeling/architectures/picodet.py
@@ -75,7 +75,7 @@ class PicoDet(BaseArch):
     def get_loss(self, ):
         loss = {}
 
-        head_outs = self._forward()
+        head_outs, _ = self._forward()
         loss_gfl = self.head.get_loss(head_outs, self.inputs)
         loss.update(loss_gfl)
         total_loss = paddle.add_n(list(loss.values()))
-- 
GitLab