We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1eac1d1 commit 296494aCopy full SHA for 296494a
1 file changed
FasterTransformer/v2/benchmark.sh
@@ -0,0 +1,12 @@
1
+SEQ_LEN=(10 20 40 60 80 100 200 300 400 500)
2
+BATCH_SIZE=(1 20)
3
+
4
+rm log.txt
5
+for batch_size in ${BATCH_SIZE[*]}
6
+do
7
+ for seq_len in ${SEQ_LEN[*]}
8
+ do
9
+ python encoder_sample.py -time 1 -batch=${batch_size} -s=${seq_len}
10
+ done
11
+done
12
0 commit comments