support deformable_detr amp (#8011)

a18c8113 · LokeZhou · GitHub · a08a3f2d · a18c8113
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

ppdet/modeling/transformers/deformable_transformer.py ppdet/modeling/transformers/deformable_transformer.py +2 -1

未找到文件。
--- a/ppdet/modeling/transformers/deformable_transformer.py
+++ b/ppdet/modeling/transformers/deformable_transformer.py
@@ -450,7 +450,8 @@ class DeformableTransformer(nn.Layer):
            temperature=pe_temperature,
            normalize=True if position_embed_type == 'sine' else False,
            embed_type=position_embed_type,
-            offset=pe_offset)
+            offset=pe_offset,
+            eps=1e-4)

        self._reset_parameters()