config_shufflenet.json

{
    "model_dir": "models/run01/",
    "train_dataset": "/home/dan/datasets/COCO/train_shards/",
    "val_dataset": "/home/dan/datasets/COCO/val_shards/",
    "pretrained_checkpoint": "pretrained/shufflenet_v2_1.0x/model.ckpt-1661328",

    "backbone": "shufflenet",
    "depth_multiplier": 1.0,
    "weight_decay": 5e-5,
    "num_classes": 80,

    "score_threshold": 0.15, "iou_threshold": 0.6, "max_boxes_per_class": 25,
    "localization_loss_weight": 1.0, "classification_loss_weight": 2.0,

    "gamma": 2.0,
    "alpha": 0.25,

    "num_steps": 350000,
    "initial_learning_rate": 1e-4,

    "min_dimension": 640,
    "batch_size": 8,
    "image_height": 640,
    "image_width": 640
}