Update to YAML File

official/vision/beta/projects/yolo/configs/experiments/darknet53.yaml

@@ -18,7 +18,7 @@ task:
     tfds_split: 'test'
     tfds_download: True
     is_training: True
-    global_batch_size: 2
+    global_batch_size: 128
     dtype: 'float16'
   validation_data:
     tfds_name: 'imagenet_a'
@@ -29,12 +29,12 @@ task:
     dtype: 'float16'
     drop_remainder: False
 trainer:
-  train_steps: 51200000 # in the paper
-  validation_steps: 25600 # size of validation data
-  validation_interval: 150
-  steps_per_loop: 150
-  summary_interval: 150
-  checkpoint_interval: 150
+  train_steps: 800000 # in the paper
+  validation_steps: 400 # size of validation data
+  validation_interval: 10000
+  steps_per_loop: 10000
+  summary_interval: 10000
+  checkpoint_interval: 10000
   optimizer_config:
     optimizer:
       type: 'sgd'
@@ -46,8 +46,8 @@ trainer:
         initial_learning_rate: 0.1
         end_learning_rate: 0.0001
         power: 4.0 
-        decay_steps: 51136000
+        decay_steps: 799000
     warmup:
       type: 'linear'
       linear:
-        warmup_steps: 64000 #learning rate rises from 0 to 0.1 over 1000 steps
+        warmup_steps: 1000 #learning rate rises from 0 to 0.1 over 1000 steps