Skip to content

GitLab

Commit 3b45a0c6 authored by Frederick Liu's avatar Frederick Liu Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 417656108
parent 0009e79b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/benchmark/transformer_benchmark.py
View file @ 3b45a0c6
...@@ -25,8 +25,8 @@ import tensorflow as tf ...@@ -25,8 +25,8 @@ import tensorflow as tf
from official.benchmark import benchmark_wrappers from official.benchmark import benchmark_wrappers
from official.benchmark import owner_utils from official.benchmark import owner_utils
from official.benchmark.perfzero_benchmark import PerfZeroBenchmark from official.benchmark.perfzero_benchmark import PerfZeroBenchmark
from official.nlp.transformer import misc from official.legacy.transformer import misc
from official.nlp.transformer import transformer_main as transformer_main from official.legacy.transformer import transformer_main
from official.utils.flags import core as flags_core from official.utils.flags import core as flags_core
TPU_DATA_DIR = 'gs://mlcompass-data/transformer' TPU_DATA_DIR = 'gs://mlcompass-data/transformer'
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment