{
  "inputs": {
    "input.1": [
      1,
      3,
      384,
      384
    ]
  },
  "calibration_num": 100,
  "calibration_method": "ema",
  "train_batchsize": 32,
  "num_samples": 1024,
  "default_loader": {
    "dataset_path": "/mnt/datasets/ILSVRC2012/val",
    "file_extensions": [
      "jpeg",
      "jpg",
      "png",
      "JPEG"
    ],
    "preprocessings": [
      {
        "resize": {
          "mode": "torchvision",
          "size": 384,
          "interpolation": "BILINEAR"
        }
      },
      {
        "centercrop": {
          "width": 384,
          "height": 384
        }
      },
      {
        "convertColor": {
          "form": "BGR2RGB"
        }
      },
      {
        "div": {
          "x": 255
        }
      },
      {
        "normalize": {
          "mean": [
            0.5,
            0.5,
            0.5
          ],
          "std": [
            0.5,
            0.5,
            0.5
          ]
        }
      },
      {
        "transpose": {
          "axis": [
            2,
            0,
            1
          ]
        }
      },
      {
        "expandDim": {
          "axis": 0
        }
      }
    ]
  },
  "qmaster": {
    "batch_size": 16,
    "num_workers": 4,
    "train_limit": 50000,
    "val_limit": 5000,
    "train_drop_last": true,
    "val_drop_last": false,
    "device": "cuda:0",
    "epochs": 30,
    "lr": 1e-05,
    "weight_decay": 0.0001,
    "optimizer": "adamw",
    "criterion": "mse",
    "max_grad_norm": 0.5,
    "use_amp": false,
    "use_kd": true,
    "kd_loss": "mse",
    "kd_alpha": 1.0,
    "kd_temperature": 4.0,
    "encoder_mode": "no_encoder",
    "train_cpu_fp": false,
    "label_smoothing": 0.0,
    "gradient_accumulation_steps": 1,
    "warmup_epochs": 0,
    "scheduler": "cosine",
    "save_best_model": false,
    "early_stopping_patience": 5,
    "early_stopping_delta": 0.0,
    "freeze_bn_after": 0,
    "fast_run": false,
    "stage1_mode": "fresh"
  }
}