Skip to content

Commit a61879a

Browse files
committed
Fix dist_transformer test
test=develop
1 parent 0de6811 commit a61879a

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

python/paddle/fluid/tests/unittests/dist_transformer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1159,6 +1159,7 @@ def prepare_encoder(src_word,
11591159
name=pos_enc_param_name,
11601160
trainable=False,
11611161
initializer=fluid.initializer.ConstantInitializer(0.001)))
1162+
str_pos_enc.stop_gradient = True
11621163
enc_input = src_word_emb + src_pos_enc
11631164
return layers.dropout(
11641165
enc_input,

0 commit comments

Comments
 (0)