From 764857224bbe1696dc7a15655e5283b9f1e1c011 Mon Sep 17 00:00:00 2001 From: Haoyu Zhang Date: Mon, 19 Aug 2019 16:26:53 -0700 Subject: [PATCH] Moving Keras ResNet models to `official/vision/image_classification` and benchmarks to `official/benchmark`. PiperOrigin-RevId: 264268533 --- tensorflow/compiler/jit/tests/auto_clustering_test.cc | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tensorflow/compiler/jit/tests/auto_clustering_test.cc b/tensorflow/compiler/jit/tests/auto_clustering_test.cc index 2154e371e83..c4db4b082ad 100644 --- a/tensorflow/compiler/jit/tests/auto_clustering_test.cc +++ b/tensorflow/compiler/jit/tests/auto_clustering_test.cc @@ -45,8 +45,8 @@ class AutoClusteringTestImpl : public AutoClusteringTest { TEST_F(AutoClusteringTestImpl, KerasImagenetMain) { // Generated from // - // bazel run -c opt --config=cuda \ - // tensorflow_models/official/resnet/keras:keras_imagenet_main \ + // TARGET_PATH=tensorflow_models/official/vision/image_classification \ + // bazel run -c opt --config=cuda ${TARGET_PATH}:resnet_imagenet_main \ // -- --skip_eval --num_gpus=1 --dtype=fp16 --batch_size=192 \ // --train_steps=210 --enable_xla --enable_eager=true // @@ -57,8 +57,8 @@ TEST_F(AutoClusteringTestImpl, KerasImagenetMain) { TEST_F(AutoClusteringTestImpl, KerasImagenetMainGraphMode) { // Generated from // - // bazel run -c opt --config=cuda \ - // tensorflow_models/official/resnet/keras:keras_imagenet_main \ + // TARGET_PATH=tensorflow_models/official/vision/image_classification \ + // bazel run -c opt --config=cuda ${TARGET_PATH}:resnet_imagenet_main \ // -- --use_synthetic_data --num_gpus=1 --batch_size=117 --train_steps=600 \ // --skip_eval=True --logtostderr --enable_xla TF_ASSERT_OK(