configs/train_2gpu_4090d.json

{
  "train_image_dir": "data_pseudo/train/images",
  "train_mask_dir": "data_pseudo/train/masks",
  "val_image_dir": "data_pseudo/val/images",
  "val_mask_dir": "data_pseudo/val/masks",
  "num_classes": 2,
  "backbone_name": "dinov3_vitl16",
  "pretrained": true,
  "backbone_weights": "dinov3_vitl16_pretrain_sat493m-eadcf0ff.pth",
  "weight_search_roots": [
    "D:/DINOv3_pretrained_weights",
    "D:/DINOv3预训练权重",
    "D:/DINOv3预训练模型/ViT models pretrained on satellite dataset (SAT-493M)",
    "D:/CBG/CoastGPT_cui",
    "D:/DINOv3预训练权重/",
    "dinov3-main"
  ],
  "use_4channel": true,
  "image_size": 256,
  "batch_size": 10,
  "grad_accum_steps": 1,
  "num_epochs": 500,
  "backbone_lr": 1e-5,
  "decoder_lr": 1e-4,
  "weight_decay": 1e-4,
  "scheduler": "cosine",
  "grad_clip": 1.0,
  "amp": true,
  "focal_alpha": [1.0, 3.0],
  "focal_gamma": 2,
  "dice_weight": 0.5,
  "focal_weight": 1.0,
  "background_weight": 1.0,
  "foreground_weight": 3.0,
  "num_workers": 4,
  "output_dir": "outputs/seaweed_dinov3_vitl16_2gpu",
  "plot_interval": 10,
  "seed": 42,
  "freeze_backbone": false,
  "use_data_augmentation": true
}