float -> float32

8afb2e6e · suweiyue · 47e9d6a1 · 8afb2e6e
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

examples/erniesage/models/ernie_model/ernie.py examples/erniesage/models/ernie_model/ernie.py +1 -1

未找到文件。
--- a/examples/erniesage/models/ernie_model/ernie.py
+++ b/examples/erniesage/models/ernie_model/ernie.py
@@ -104,7 +104,7 @@ class ErnieModel(object):
        zero = L.fill_constant([1], dtype='int64', value=0)
        input_mask = L.logical_not(L.equal(src_ids,
                                           zero))  # assume pad id == 0
-        input_mask = L.cast(input_mask, 'float')
+        input_mask = L.cast(input_mask, 'float32')
        input_mask.stop_gradient = True
        return input_mask