maxmustermann/DeepLearningExamples
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

fix typo (#605)

Browse source
This commit is contained in:
372046933 2020-07-21 23:10:01 +08:00 committed by GitHub
parent 67370c8948
commit 108a783a77
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
FasterTransformer/v2.1/sample/tensorflow/encoder_sample.py
View file

@ -14,7 +14,7 @@
'''
This is a sample code to demonstrate how to use the TensorFlow custom op with
FasterTransformer library in decoder.
FasterTransformer library in encoder.
This sample code builds a BERT transformer model by TensorFlow and TensorFlow
custom op. Then compare the maximum difference of them to verify the correctness

4
FasterTransformer/v2/sample/tensorflow/encoder_sample.py
View file

@ -105,5 +105,5 @@ if __name__ == "__main__":
tf_time = time_test(sess, tf_encoder_result, ite)
op_time = time_test(sess, op_encoder_result, ite)
print("[INFO] TF decoder time costs: {} ms".format(tf_time))
print("[INFO] OP decoder time costs: {} ms".format(op_time))
print("[INFO] TF encoder time costs: {} ms".format(tf_time))
print("[INFO] OP encoder time costs: {} ms".format(op_time))