Skip to content

GitLab

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

Internal change 

PiperOrigin-RevId: 346436382
parent 77d47f79
Show 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 @ a49ea856
......@@ -41,10 +41,12 @@ task:
stem_pool_temporal_stride: 2
dropout_rate: 0.5
train_data:
name: kinetics400
global_batch_size: 1024
dtype: 'bfloat16'
shuffle_buffer_size: 1024
validation_data:
name: kinetics400
global_batch_size: 32
dtype: 'bfloat16'
drop_remainder: false
......
official/vision/beta/configs/experiments/video_classification/k400_slowonly8x8_tpu.yaml
View file @ a49ea856
......@@ -41,6 +41,7 @@ task:
stem_pool_temporal_stride: 1
dropout_rate: 0.5
train_data:
name: kinetics400
feature_shape: !!python/tuple
- 8
- 224
......@@ -51,6 +52,7 @@ task:
dtype: 'bfloat16'
shuffle_buffer_size: 1024
validation_data:
name: kinetics400
global_batch_size: 32
dtype: 'bfloat16'
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