Skip to content

GitLab

Commit 5ee7a462 authored by Yeqing Li's avatar Yeqing Li Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 346436382
parent a99dc30d
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
official/vision/beta/configs/experiments/video_classification/k400_3d-resnet50_tpu.yaml
View file @ 5ee7a462
...@@ -41,10 +41,12 @@ task: ...@@ -41,10 +41,12 @@ task:
stem_pool_temporal_stride: 2 stem_pool_temporal_stride: 2
dropout_rate: 0.5 dropout_rate: 0.5
train_data: train_data:
name: kinetics400
global_batch_size: 1024 global_batch_size: 1024
dtype: 'bfloat16' dtype: 'bfloat16'
shuffle_buffer_size: 1024 shuffle_buffer_size: 1024
validation_data: validation_data:
name: kinetics400
global_batch_size: 32 global_batch_size: 32
dtype: 'bfloat16' dtype: 'bfloat16'
drop_remainder: false drop_remainder: false
......
official/vision/beta/configs/experiments/video_classification/k400_slowonly8x8_tpu.yaml
View file @ 5ee7a462
...@@ -41,6 +41,7 @@ task: ...@@ -41,6 +41,7 @@ task:
stem_pool_temporal_stride: 1 stem_pool_temporal_stride: 1
dropout_rate: 0.5 dropout_rate: 0.5
train_data: train_data:
name: kinetics400
feature_shape: !!python/tuple feature_shape: !!python/tuple
- 8 - 8
- 224 - 224
...@@ -51,6 +52,7 @@ task: ...@@ -51,6 +52,7 @@ task:
dtype: 'bfloat16' dtype: 'bfloat16'
shuffle_buffer_size: 1024 shuffle_buffer_size: 1024
validation_data: validation_data:
name: kinetics400
global_batch_size: 32 global_batch_size: 32
dtype: 'bfloat16' dtype: 'bfloat16'
drop_remainder: false drop_remainder: false
......
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