config.json

{
  "architectures": [
    "CetaceanClassifierModelForImageClassification"
  ],
  "aug": {
    "crop_l": 0.75,
    "crop_r": 1.3333333333333333,
    "crop_scale": 0.9,
    "p_affine": 0.5,
    "p_blur": 0.05,
    "p_bright_contrast": 0.5,
    "p_cutout": 0.05,
    "p_downscale": 0.0,
    "p_gray": 0.1,
    "p_noise": 0.05,
    "p_posterize": 0.2,
    "p_rain": 0.05,
    "p_shuffle": 0.3,
    "p_snow": 0.1,
    "rotate": 15,
    "shear": 3,
    "translate": 0.25
  },
  "auto_map": {
    "AutoConfig": "configuration_cetacean_classifier.CetaceanClassifierConfig",
    "AutoModelForImageClassification": "modeling_cetacean_classifier.CetaceanClassifierModelForImageClassification"
  },
  "batch_size": 8,
  "bbox_conf_threshold": 0.01,
  "bboxes": {
    "backfin": 0.15,
    "detic": 0.05,
    "fullbody": 0.6,
    "fullbody_charm": 0.15,
    "none": 0.05
  },
  "global_pool": {
    "arch": "GeM",
    "p": 3,
    "train": false
  },
  "image_size": [
    768,
    768
  ],
  "loss_fn": "CrossEntropy",
  "loss_id_ratio": 0.437338,
  "lr_backbone": 0.0016,
  "lr_decay_scale": 0.01,
  "lr_head": 0.016,
  "margin_coef_id": 0.27126,
  "margin_coef_species": 0.226253,
  "margin_cons_id": 0.05,
  "margin_cons_species": 0.05,
  "margin_power_id": -0.364399,
  "margin_power_species": -0.720133,
  "max_epochs": 30,
  "model_name": "tf_efficientnetv2_m",
  "model_type": "cetaceanet",
  "n_center_id": 2,
  "n_center_species": 2,
  "n_data": -1,
  "n_splits": -1,
  "normalization": "batchnorm",
  "num_classes": 15587,
  "num_species_classes": 26,
  "optimizer": "AdamW",
  "out_indices": [
    3,
    4
  ],
  "pretrained": true,
  "pseudo_conf_threshold": 0.6,
  "pseudo_label": "pseudo_labels/round2.csv",
  "s_id": 20.9588,
  "s_species": 33.1383,
  "test_bboxes": [
    "fullbody",
    "fullbody_charm"
  ],
  "torch_dtype": "float32",
  "transformers_version": "4.46.0",
  "val_bbox": "fullbody",
  "warmup_steps_ratio": 0.2
}