From 0f9f93e57bcb8c300cd91095acb22bccea795651 Mon Sep 17 00:00:00 2001
From: wangxinxin08 <69842442+wangxinxin08@users.noreply.github.com>
Date: Mon, 11 Apr 2022 19:53:02 +0800
Subject: [PATCH] remove register_buffer (#5632)

---
 ppdet/modeling/heads/ppyoloe_head.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ppdet/modeling/heads/ppyoloe_head.py b/ppdet/modeling/heads/ppyoloe_head.py
index 31e7590a0..01d7e9174 100644
--- a/ppdet/modeling/heads/ppyoloe_head.py
+++ b/ppdet/modeling/heads/ppyoloe_head.py
@@ -132,8 +132,8 @@ class PPYOLOEHead(nn.Layer):
 
         if self.eval_size:
             anchor_points, stride_tensor = self._generate_anchors()
-            self.register_buffer('anchor_points', anchor_points)
-            self.register_buffer('stride_tensor', stride_tensor)
+            self.anchor_points = anchor_points
+            self.stride_tensor = stride_tensor
 
     def forward_train(self, feats, targets):
         anchors, anchor_points, num_anchors_list, stride_tensor = \
-- 
GitLab