model task

official/vision/beta/projects/yolo/configs/experiments/yolov4-csp/inference/640.yaml

-runtime:
-  distribution_strategy: 'mirrored'
-  mixed_precision_dtype: 'float16'
-  num_gpus: 1
-task:
-  init_checkpoint: /home/vbanna/Research/checkpoints/yolo/v4-csp
-  init_checkpoint_modules: ['backbone', 'decoder']
-  smart_bias_lr: 0.1
-  model:
-    darknet_based_model: False
-    input_size: [640, 640, 3]
-    backbone:
-      type: 'darknet'
-      darknet:
-        model_id: 'altered_cspdarknet53'
-        max_level: 5
-        min_level: 3
-    decoder:
-      type: yolo_decoder
-      yolo_decoder:
-        version: v4
-        type: csp
-    head:
-      smart_bias: true
-    detection_generator:
-      box_type:
-        'all': scaled
-      scale_xy:
-        'all': 2.0
-      max_boxes: 300
-      nms_type: greedy
-      iou_thresh: 0.25
-      nms_thresh: 0.45
-      pre_nms_points: 500
-    loss:
-      use_scaled_loss: true
-      update_on_repeat: true
-      box_loss_type:  
-        'all': ciou
-      ignore_thresh:
-        'all': 0.7
-      iou_normalizer: 
-        'all': 0.05
-      cls_normalizer: 
-        'all': 0.3
-      object_normalizer: 
-        '5': 0.28
-        '4': 0.70
-        '3': 2.80
-      objectness_smooth: 
-        'all': 1.0
-    norm_activation:
-      activation: leaky
-      norm_epsilon: 0.0001
-      norm_momentum: 0.97
-      use_sync_bn: false
-    num_classes: 80
-    anchor_boxes:
-      anchors_per_scale: 3
-      boxes: [box: [12, 16],   box: [19, 36],   box: [40, 28], 
-              box: [36, 75],   box: [76, 55],   box: [72, 146], 
-              box: [142, 110], box: [192, 243], box: [459, 401]]
-  train_data:
-    global_batch_size: 1
-    input_path:  '/media/vbanna/DATA_SHARE/CV/datasets/COCO_raw/records/train*'
-    shuffle_buffer_size: 1
-    parser:
-      mosaic: 
-        mosaic_frequency: 1.0
-        mixup_frequency: 0.0
-        mosaic_crop_mode: 'scale'
-        mosaic_center: 0.25
-        aug_scale_min: 0.1
-        aug_scale_max: 1.9
-      letter_box: True
-      random_flip: True
-      aug_rand_translate: 0.1
-      area_thresh: 0.1
-  validation_data:
-    global_batch_size: 1
-    input_path: '/media/vbanna/DATA_SHARE/CV/datasets/COCO_raw/records/val*'
-trainer:
-  steps_per_loop: 1
-  summary_interval: 1
-  optimizer_config:
-    ema: null
\ No newline at end of file