Skip to content

GitLab

Commit fa5b66e5 authored by Jing Li's avatar Jing Li Committed by A. Unique TensorFlower
Browse files

support TPU on resnet50 using keras compile/fit 

PiperOrigin-RevId: 273358759
parent 482784ab
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/vision/image_classification/resnet_imagenet_main.py
View file @ fa5b66e5
...@@ -83,7 +83,8 @@ def run(flags_obj): ...@@ -83,7 +83,8 @@ def run(flags_obj):
num_gpus=flags_obj.num_gpus, num_gpus=flags_obj.num_gpus,
num_workers=num_workers, num_workers=num_workers,
all_reduce_alg=flags_obj.all_reduce_alg, all_reduce_alg=flags_obj.all_reduce_alg,
num_packs=flags_obj.num_packs) num_packs=flags_obj.num_packs,
tpu_address=flags_obj.tpu)
if strategy: if strategy:
# flags_obj.enable_get_next_as_optional controls whether enabling # flags_obj.enable_get_next_as_optional controls whether enabling
......
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