Updates default learning rate for UCF-101 experiment.

PiperOrigin-RevId: 375725859

Updates default learning rate for UCF-101 experiment.
PiperOrigin-RevId: 375725859
3d03e675 · Yeqing Li · A. Unique TensorFlower · cee98c90 · 3d03e675
Commit 3d03e675 authored May 25, 2021 by Yeqing Li Committed by A. Unique TensorFlower May 25, 2021
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

official/vision/beta/configs/video_classification.py official/vision/beta/configs/video_classification.py +6 -1

No files found.
--- a/official/vision/beta/configs/video_classification.py
+++ b/official/vision/beta/configs/video_classification.py
@@ -254,7 +254,12 @@ def video_classification_ucf101() -> cfg.ExperimentConfig:
          'task.validation_data.is_training != None',
          'task.train_data.num_classes == task.validation_data.num_classes',
      ])
-  add_trainer(config, train_batch_size=64, eval_batch_size=16, train_epochs=100)
+  add_trainer(
+      config,
+      train_batch_size=64,
+      eval_batch_size=16,
+      learning_rate=0.8,
+      train_epochs=100)
  return config