codecs/segmentation_map/config.json

{
  "begin_cosine_annealing": 38000,
  "decoder_input_dim": null,
  "ema_decay": 0.9999,
  "ema_model_weights": true,
  "ema_update_freq": 1,
  "embedding_dim": 4,
  "encoder_output_dim": 512,
  "levels": [
    8,
    5,
    5,
    5
  ],
  "lr": 7.5e-05,
  "lr_cosine_period": 76000,
  "lr_warmup": 1000,
  "num_downsamples": 3,
  "num_res_layers": 2,
  "output_activation_extension": null,
  "quantisation_loss_weight": 1.0,
  "res_hidden_dims": 512,
  "variational": false
}