Commit 8dc7f8f1 authored by Yeqing Li's avatar Yeqing Li Committed by A. Unique TensorFlower
Browse files

Internal change

PiperOrigin-RevId: 348063144
parent f03e5f9d
# 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: 42104 decay_steps: 42000
warmup: warmup:
linear: linear:
warmup_steps: 1053 warmup_steps: 1050
train_steps: 42104 train_steps: 42000
steps_per_loop: 500 steps_per_loop: 500
summary_interval: 500 summary_interval: 500
validation_interval: 500 validation_interval: 500
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