Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
b7c8dab7
Commit
b7c8dab7
authored
Aug 13, 2019
by
A. Unique TensorFlower
Browse files
Merge pull request #7439 from tfboyd:lstm_no_xla_8_gpu
PiperOrigin-RevId: 263261200
parents
c789d64b
0d92f34a
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
1 addition
and
16 deletions
+1
-16
official/staging/shakespeare/shakespeare_benchmark.py
official/staging/shakespeare/shakespeare_benchmark.py
+0
-15
research/lstm_object_detection/tflite/mobile_ssd_tflite_client.h
...h/lstm_object_detection/tflite/mobile_ssd_tflite_client.h
+1
-1
No files found.
official/staging/shakespeare/shakespeare_benchmark.py
View file @
b7c8dab7
...
@@ -208,21 +208,6 @@ class ShakespeareAccuracy(ShakespeareBenchmarkBase):
...
@@ -208,21 +208,6 @@ class ShakespeareAccuracy(ShakespeareBenchmarkBase):
FLAGS
.
model_dir
=
''
FLAGS
.
model_dir
=
''
self
.
_run_and_report_benchmark
()
self
.
_run_and_report_benchmark
()
def
benchmark_xla_8_gpu
(
self
):
"""Benchmark 8 gpu w/xla.
This is test is for accuracy not scaling. The batch-size is not scaled to
the number of gpus.
"""
self
.
_setup
()
FLAGS
.
num_gpus
=
8
FLAGS
.
training_data
=
self
.
train_data
FLAGS
.
batch_size
=
64
FLAGS
.
train_epochs
=
43
FLAGS
.
model_dir
=
''
FLAGS
.
enable_xla
=
True
self
.
_run_and_report_benchmark
()
class
ShakespeareKerasBenchmarkReal
(
ShakespeareBenchmarkBase
):
class
ShakespeareKerasBenchmarkReal
(
ShakespeareBenchmarkBase
):
"""Benchmark accuracy tests."""
"""Benchmark accuracy tests."""
...
...
research/lstm_object_detection/tflite/mobile_ssd_tflite_client.h
View file @
b7c8dab7
...
@@ -26,7 +26,7 @@ limitations under the License.
...
@@ -26,7 +26,7 @@ limitations under the License.
#include "mobile_ssd_client.h"
#include "mobile_ssd_client.h"
#include "protos/anchor_generation_options.pb.h"
#include "protos/anchor_generation_options.pb.h"
#ifdef ENABLE_EDGETPU
#ifdef ENABLE_EDGETPU
#include "libedgetpu/edgetpu.h"
#include "libedgetpu/
lib
edgetpu.h"
#endif // ENABLE_EDGETPU
#endif // ENABLE_EDGETPU
namespace
lstm_object_detection
{
namespace
lstm_object_detection
{
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment