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
8dc7f8f1
Commit
8dc7f8f1
authored
Dec 17, 2020
by
Yeqing Li
Committed by
A. Unique TensorFlower
Dec 17, 2020
Browse files
Internal change
PiperOrigin-RevId: 348063144
parent
f03e5f9d
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
10 additions
and
4 deletions
+10
-4
official/vision/beta/configs/experiments/video_classification/k400_3d-resnet50_tpu.yaml
...xperiments/video_classification/k400_3d-resnet50_tpu.yaml
+10
-4
No files found.
official/vision/beta/configs/experiments/video_classification/k400_3d-resnet50_tpu.yaml
View file @
8dc7f8f1
# 3D ResNet-50 video classification on Kinetics-400. 75.3% top-1 and 91.2% top-5 accuracy.
# 3D ResNet-50 video classification on Kinetics-400. 75.3% top-1 and 91.2% top-5 accuracy.
#
# --experiment_type=video_classification_kinetics400
# Expected accuracy on TPU 8x8: 75.1%
# Updated: 2020-12-16
runtime
:
runtime
:
distribution_strategy
:
'
tpu'
distribution_strategy
:
'
tpu'
mixed_precision_dtype
:
'
bfloat16'
mixed_precision_dtype
:
'
bfloat16'
task
:
task
:
model
:
model
:
dropout_rate
:
0.5
norm_activation
:
use_sync_bn
:
false
backbone
:
backbone
:
resnet_3d
:
resnet_3d
:
block_specs
:
!!python/tuple
block_specs
:
!!python/tuple
...
@@ -39,7 +46,6 @@ task:
...
@@ -39,7 +46,6 @@ task:
stem_conv_temporal_kernel_size
:
5
stem_conv_temporal_kernel_size
:
5
stem_conv_temporal_stride
:
2
stem_conv_temporal_stride
:
2
stem_pool_temporal_stride
:
2
stem_pool_temporal_stride
:
2
dropout_rate
:
0.5
train_data
:
train_data
:
name
:
kinetics400
name
:
kinetics400
global_batch_size
:
1024
global_batch_size
:
1024
...
@@ -55,11 +61,11 @@ trainer:
...
@@ -55,11 +61,11 @@ trainer:
learning_rate
:
learning_rate
:
cosine
:
cosine
:
initial_learning_rate
:
0.8
initial_learning_rate
:
0.8
decay_steps
:
42
104
decay_steps
:
42
000
warmup
:
warmup
:
linear
:
linear
:
warmup_steps
:
105
3
warmup_steps
:
105
0
train_steps
:
42
104
train_steps
:
42
000
steps_per_loop
:
500
steps_per_loop
:
500
summary_interval
:
500
summary_interval
:
500
validation_interval
:
500
validation_interval
:
500
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