chillax/main.json

{
  "meta": {
    "name": "chillax-selfsupervised"
  },
  "evaluators": [
    {
      "name": "accuracy"
    },
    {
      "name": "hierarchical"
    },
    {
      "name": "topk_accuracy"
    }
  ],
  "with_wordnet": true,
  "interactor": {
    "name": "oracle"
  },
  "observers": [
    {
      "name": "stream"
    },
    {
      "name": "json"
    }
  ],
  "runner": {
    "name": "epoch",
    "epochs": 10
  },
  "dataset": {
    "name": "nabirds",
    "base_path": "/home/brust/datasets/nabirds",
    "side_length": 512,
    "use_lazy_mode": true
  },
  "model": {
    "classifier": {
      "name": "chillax",
      "l2": 5e-5,
      "mlnp": true,
      "normalize_scores": true
    },
    "base_model": {
      "name": "keras",
      "augmentation": {
        "do_random_flip_vertical": false,
        "do_random_scale": false,
        "do_random_rotate": false,
        "do_random_brightness_and_contrast": false,
        "do_random_hue_and_saturation": false,
        "do_random_crop": false
      },
      "trainer": {
        "name": "fast_single_shot",
        "batch_size": 16,
        "sequential_training_batches": 2,
        "inner_steps": 5976
      },
      "feature_extractor": {
        "side_length": 448,
        "trainable": true,
        "architecture": "ResNet50V2",
        "l2": 5e-5,
        "use_pretrained_weights": "inat_features.h5"
      },
      "preprocessor": {
        "random_crop_to_size": [
          448,
          448
        ],
        "channel_mean": [
          125.30513277,
          129.66606421,
          118.45121113
        ],
        "channel_stddev": [
          57.0045467,
          56.70059436,
          68.44430446
        ]
      },
      "optimizer": {
        "name": "sgd",
        "momentum": 0.9
      },
      "learning_rate_schedule": {
        "name": "sgdr",
        "maximum_lr": 0.0044,
        "T_0": 59760,
        "T_mult": 1,
        "minimum_lr": 1e-06,
        "warmup_steps": 747,
        "warmup_lr": 0.01
      }
    }
  }
}