未验证 提交 0f9f93e5 编写于 作者: W wangxinxin08 提交者: GitHub

remove register_buffer (#5632)

上级 2ef30b49
...@@ -132,8 +132,8 @@ class PPYOLOEHead(nn.Layer): ...@@ -132,8 +132,8 @@ class PPYOLOEHead(nn.Layer):
if self.eval_size: if self.eval_size:
anchor_points, stride_tensor = self._generate_anchors() anchor_points, stride_tensor = self._generate_anchors()
self.register_buffer('anchor_points', anchor_points) self.anchor_points = anchor_points
self.register_buffer('stride_tensor', stride_tensor) self.stride_tensor = stride_tensor
def forward_train(self, feats, targets): def forward_train(self, feats, targets):
anchors, anchor_points, num_anchors_list, stride_tensor = \ anchors, anchor_points, num_anchors_list, stride_tensor = \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册