|
{ |
|
"_name_or_path": "facebook/dinov2-large", |
|
"apply_layernorm": true, |
|
"architectures": [ |
|
"NewheadDinov2ForImageClassification" |
|
], |
|
"attention_probs_dropout_prob": 0.0, |
|
"drop_path_rate": 0.0, |
|
"hidden_act": "gelu", |
|
"hidden_dropout_prob": 0.0, |
|
"hidden_size": 1024, |
|
"id2label": { |
|
"0": "Acropore_branched", |
|
"1": "Acropore_digitised", |
|
"2": "Acropore_sub_massive", |
|
"3": "Acropore_tabular", |
|
"4": "Algae_assembly", |
|
"5": "Algae_drawn_up", |
|
"6": "Algae_limestone", |
|
"7": "Algae_sodding", |
|
"8": "Atra/Leucospilota", |
|
"9": "Bleached_coral", |
|
"10": "Blurred", |
|
"11": "Dead_coral", |
|
"12": "Fish", |
|
"13": "Homo_sapiens", |
|
"14": "Human_object", |
|
"15": "Living_coral", |
|
"16": "Millepore", |
|
"17": "No_acropore_encrusting", |
|
"18": "No_acropore_foliaceous", |
|
"19": "No_acropore_massive", |
|
"20": "No_acropore_solitary", |
|
"21": "No_acropore_sub_massive", |
|
"22": "Rock", |
|
"23": "Rubble", |
|
"24": "Sand", |
|
"25": "Sea_cucumber", |
|
"26": "Sea_urchins", |
|
"27": "Sponge", |
|
"28": "Syringodium_isoetifolium", |
|
"29": "Thalassodendron_ciliatum", |
|
"30": "Useless" |
|
}, |
|
"image_size": 518, |
|
"initializer_range": 0.02, |
|
"label2id": { |
|
"Acropore_branched": 0, |
|
"Acropore_digitised": 1, |
|
"Acropore_sub_massive": 2, |
|
"Acropore_tabular": 3, |
|
"Algae_assembly": 4, |
|
"Algae_drawn_up": 5, |
|
"Algae_limestone": 6, |
|
"Algae_sodding": 7, |
|
"Atra/Leucospilota": 8, |
|
"Bleached_coral": 9, |
|
"Blurred": 10, |
|
"Dead_coral": 11, |
|
"Fish": 12, |
|
"Homo_sapiens": 13, |
|
"Human_object": 14, |
|
"Living_coral": 15, |
|
"Millepore": 16, |
|
"No_acropore_encrusting": 17, |
|
"No_acropore_foliaceous": 18, |
|
"No_acropore_massive": 19, |
|
"No_acropore_solitary": 20, |
|
"No_acropore_sub_massive": 21, |
|
"Rock": 22, |
|
"Rubble": 23, |
|
"Sand": 24, |
|
"Sea_cucumber": 25, |
|
"Sea_urchins": 26, |
|
"Sponge": 27, |
|
"Syringodium_isoetifolium": 28, |
|
"Thalassodendron_ciliatum": 29, |
|
"Useless": 30 |
|
}, |
|
"layer_norm_eps": 1e-06, |
|
"layerscale_value": 1.0, |
|
"mlp_ratio": 4, |
|
"model_type": "dinov2", |
|
"num_attention_heads": 16, |
|
"num_channels": 3, |
|
"num_hidden_layers": 24, |
|
"out_features": [ |
|
"stage24" |
|
], |
|
"out_indices": [ |
|
24 |
|
], |
|
"patch_size": 14, |
|
"problem_type": "multi_label_classification", |
|
"qkv_bias": true, |
|
"reshape_hidden_states": true, |
|
"stage_names": [ |
|
"stem", |
|
"stage1", |
|
"stage2", |
|
"stage3", |
|
"stage4", |
|
"stage5", |
|
"stage6", |
|
"stage7", |
|
"stage8", |
|
"stage9", |
|
"stage10", |
|
"stage11", |
|
"stage12", |
|
"stage13", |
|
"stage14", |
|
"stage15", |
|
"stage16", |
|
"stage17", |
|
"stage18", |
|
"stage19", |
|
"stage20", |
|
"stage21", |
|
"stage22", |
|
"stage23", |
|
"stage24" |
|
], |
|
"torch_dtype": "float32", |
|
"transformers_version": "4.41.1", |
|
"use_swiglu_ffn": false, |
|
"initial_learning_rate": 0.001, |
|
"train_batch_size": 32, |
|
"eval_batch_size": 32, |
|
"optimizer": { |
|
"type": "Adam" |
|
}, |
|
"lr_scheduler_type": { |
|
"type": "ReduceLROnPlateau" |
|
}, |
|
"patience_lr_scheduler": 5, |
|
"factor_lr_scheduler": 0.1, |
|
"weight_decay": 0.0001, |
|
"early_stopping_patience": 10, |
|
"freeze_encoder": true, |
|
"data_augmentation": true, |
|
"num_epochs": 150, |
|
"emissions_data": { |
|
"emissions": 1.414263264227963, |
|
"source": "Code Carbon", |
|
"training_type": "fine-tuning", |
|
"geographical_location": "Brest, France", |
|
"hardware_used": "NVIDIA Tesla V100 PCIe 32 Go" |
|
} |
|
} |