CoolFace
Apppublic

santosh3110/Ecoclassify-Wildlife_Classifier

sourceHugging Faceupdated 1y agoView on Hugging Face
0likes
params.yaml65 linesDownload Raw Back to root
1# Image settings2IMAGE_SIZE: [224, 224]3BATCH_SIZE: 324RANDOM_SEED: 425 6# Training hyperparameters7CUSTOMCNN:8  dropout: 0.59  hidden_units: 51210  num_classes: 811 12EPOCHS: 10013LEARNING_RATE: 0.000114WEIGHT_DECAY: 0.000515ALPHA_L2SP: 1e-4  16 17# Scheduler18SCHEDULER_PATIENCE: 219SCHEDULER_FACTOR: 0.520 21# Model choice 22MODEL_TO_USE: "Resnet50"23 24# Label smoothing25LABEL_SMOOTHING: 0.126 27# Early stopping28EARLY_STOPPING_PATIENCE: 1029 30# Gradient clipping31GRAD_CLIP: 1.032 33# Temperature scaling for calibration34TEMPERATURE_TUNING:35  ENABLED: true36  SEARCH_RANGE: [0.5, 2.0]37  SEARCH_STEPS: 2038 39EVALUATION:40  TOP_K: 341  USE_TEST_SET: false42 43GRADCAM:44  gradcam_target_layer: "layer4" 45  num_images: 2046 47BATCH_INFERENCE:48  batch_size: 3249  num_workers: 250  top_k: 351 52FINE_TUNING:53  batch_size: 3254  epochs: 555  unfreeze_backbone: false56  patience: 557  learning_rate: 0.00158  scheduler_patience: 259  scheduler_factor: 0.560  crop_size: 22461  flip: true62  brightness: 0.263  contrast: 0.264  saturation: 0.265  hue: 0.1