Skip to content

Commit 3b45a0c

Browse files
Internal change
PiperOrigin-RevId: 417656108
1 parent 0009e79 commit 3b45a0c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

official/benchmark/transformer_benchmark.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,8 @@
2525
from official.benchmark import benchmark_wrappers
2626
from official.benchmark import owner_utils
2727
from official.benchmark.perfzero_benchmark import PerfZeroBenchmark
28-
from official.nlp.transformer import misc
29-
from official.nlp.transformer import transformer_main as transformer_main
28+
from official.legacy.transformer import misc
29+
from official.legacy.transformer import transformer_main
3030
from official.utils.flags import core as flags_core
3131

3232
TPU_DATA_DIR = 'gs://mlcompass-data/transformer'

0 commit comments

Comments
 (0)