372046933 5 ani în urmă
părinte
comite
108a783a77

+ 1 - 1
FasterTransformer/v2.1/sample/tensorflow/encoder_sample.py

@@ -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

+ 2 - 2
FasterTransformer/v2/sample/tensorflow/encoder_sample.py

@@ -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))