Internal change

PiperOrigin-RevId: 350257408

official/vision/beta/configs/experiments/video_classification/k400_slowonly8x8_tpu.yaml

@@ -53,6 +53,13 @@ task:
     shuffle_buffer_size: 1024
   validation_data:
     name: kinetics400
+    feature_shape: !!python/tuple
+    - 8
+    - 256
+    - 256
+    - 3
+    temporal_stride: 8
+    num_test_clips: 1
     global_batch_size: 32
     dtype: 'bfloat16'
     drop_remainder: false

official/vision/beta/modeling/backbones/resnet_3d.py

@@ -294,6 +294,10 @@ def build_resnet3d(
       temporal_kernel_sizes=temporal_kernel_sizes,
       use_self_gating=use_self_gating,
       input_specs=input_specs,
+      stem_conv_temporal_kernel_size=backbone_cfg
+      .stem_conv_temporal_kernel_size,
+      stem_conv_temporal_stride=backbone_cfg.stem_conv_temporal_stride,
+      stem_pool_temporal_stride=backbone_cfg.stem_pool_temporal_stride,
       activation=norm_activation_config.activation,
       use_sync_bn=norm_activation_config.use_sync_bn,
       norm_momentum=norm_activation_config.norm_momentum,