File size: 1,201 Bytes
dd0ae11 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 | {
"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
}
|