diff --git a/configs/picodet/README.md b/configs/picodet/README.md
index b4df109fef82582796b1b5bb5d4a89d8035800c5..fb7cb426119697730dab854928675e7aa4e0fc4b 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 9c67005b0aee1bddf27b0a8cda1c3155a64f3c25..cd807a9fa5fe05a3f5d88363d8cb59a06af15e7e 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()))