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
ee0b6a6d
Commit
ee0b6a6d
authored
Apr 16, 2020
by
Hongkun Yu
Committed by
A. Unique TensorFlower
Apr 16, 2020
Browse files
Clean up configs.
PiperOrigin-RevId: 306883491
parent
1ff3c319
Changes
6
Show whitespace changes
Inline
Side-by-side
Showing
6 changed files
with
2 additions
and
14 deletions
+2
-14
official/vision/image_classification/configs/examples/efficientnet/imagenet/efficientnet-b0-gpu.yaml
...s/examples/efficientnet/imagenet/efficientnet-b0-gpu.yaml
+0
-2
official/vision/image_classification/configs/examples/efficientnet/imagenet/efficientnet-b0-tpu.yaml
...s/examples/efficientnet/imagenet/efficientnet-b0-tpu.yaml
+0
-2
official/vision/image_classification/configs/examples/efficientnet/imagenet/efficientnet-b1-gpu.yaml
...s/examples/efficientnet/imagenet/efficientnet-b1-gpu.yaml
+0
-2
official/vision/image_classification/configs/examples/efficientnet/imagenet/efficientnet-b1-tpu.yaml
...s/examples/efficientnet/imagenet/efficientnet-b1-tpu.yaml
+0
-2
official/vision/image_classification/configs/examples/resnet/imagenet/gpu.yaml
..._classification/configs/examples/resnet/imagenet/gpu.yaml
+1
-3
official/vision/image_classification/configs/examples/resnet/imagenet/tpu.yaml
..._classification/configs/examples/resnet/imagenet/tpu.yaml
+1
-3
No files found.
official/vision/image_classification/configs/examples/efficientnet/imagenet/efficientnet-b0-gpu.yaml
View file @
ee0b6a6d
...
...
@@ -3,8 +3,6 @@
# Reaches ~76.1% within 350 epochs.
# Note: This configuration uses a scaled per-replica batch size based on the number of devices.
runtime
:
model_dir
:
null
mode
:
'
train_and_eval'
distribution_strategy
:
'
mirrored'
num_gpus
:
1
train_dataset
:
...
...
official/vision/image_classification/configs/examples/efficientnet/imagenet/efficientnet-b0-tpu.yaml
View file @
ee0b6a6d
...
...
@@ -3,8 +3,6 @@
# Reaches ~76.1% within 350 epochs.
# Note: This configuration uses a scaled per-replica batch size based on the number of devices.
runtime
:
model_dir
:
null
mode
:
'
train_and_eval'
distribution_strategy
:
'
tpu'
train_dataset
:
name
:
'
imagenet2012'
...
...
official/vision/image_classification/configs/examples/efficientnet/imagenet/efficientnet-b1-gpu.yaml
View file @
ee0b6a6d
# Note: This configuration uses a scaled per-replica batch size based on the number of devices.
runtime
:
model_dir
:
null
mode
:
'
train_and_eval'
distribution_strategy
:
'
mirrored'
num_gpus
:
1
train_dataset
:
...
...
official/vision/image_classification/configs/examples/efficientnet/imagenet/efficientnet-b1-tpu.yaml
View file @
ee0b6a6d
...
...
@@ -2,8 +2,6 @@
# Takes ~3 minutes, 15 seconds per epoch for v3-32.
# Note: This configuration uses a scaled per-replica batch size based on the number of devices.
runtime
:
model_dir
:
null
mode
:
'
train_and_eval'
distribution_strategy
:
'
tpu'
train_dataset
:
name
:
'
imagenet2012'
...
...
official/vision/image_classification/configs/examples/resnet/imagenet/gpu.yaml
View file @
ee0b6a6d
...
...
@@ -2,8 +2,6 @@
# Reaches > 76.1% within 90 epochs.
# Note: This configuration uses a scaled per-replica batch size based on the number of devices.
runtime
:
model_dir
:
null
mode
:
'
train_and_eval'
distribution_strategy
:
'
mirrored'
num_gpus
:
1
train_dataset
:
...
...
@@ -33,7 +31,7 @@ validation_dataset:
mean_subtract
:
True
standardize
:
True
model
:
model_
name
:
'
resnet'
name
:
'
resnet'
model_params
:
rescale_inputs
:
False
optimizer
:
...
...
official/vision/image_classification/configs/examples/resnet/imagenet/tpu.yaml
View file @
ee0b6a6d
...
...
@@ -3,8 +3,6 @@
# Reaches > 76.1% within 90 epochs.
# Note: This configuration uses a scaled per-replica batch size based on the number of devices.
runtime
:
model_dir
:
null
mode
:
'
train_and_eval'
distribution_strategy
:
'
tpu'
train_dataset
:
name
:
'
imagenet2012'
...
...
@@ -35,7 +33,7 @@ validation_dataset:
standardize
:
True
dtype
:
'
bfloat16'
model
:
model_
name
:
'
resnet'
name
:
'
resnet'
model_params
:
rescale_inputs
:
False
optimizer
:
...
...
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