Skip to content

Commit d05554c

Browse files
authored
Update encoder.py
1 parent 7dc7057 commit d05554c

File tree

1 file changed

+1
-1
lines changed
  • projects/mmdet3d_plugin/bevformer/modules

1 file changed

+1
-1
lines changed

projects/mmdet3d_plugin/bevformer/modules/encoder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ def point_sampling(self, reference_points, pc_range, img_metas):
146146

147147
return reference_points_cam, bev_mask
148148

149-
# @auto_fp16() # This may cause 'grad_norm: nan'
149+
@auto_fp16() # This may cause 'grad_norm: nan'
150150
def forward(self,
151151
bev_query,
152152
key,

0 commit comments

Comments
 (0)