File size: 1,675 Bytes
dffb95f | 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 43 44 45 46 47 48 49 50 51 52 53 | {
"best_val_iou": 0.7641911350500089,
"final_train_loss": 0.028582889636827587,
"final_val_loss": 2.0107891160738878,
"final_train_iou": 0.9486552243039552,
"final_val_iou": 0.7626956336604901,
"total_epochs": 100,
"patchify_enabled": true,
"patches_per_image": 625,
"inference_stride": 64,
"config": {
"train_txt": "data-segments/train_sample.txt",
"val_txt": "data-segments/validation_sample.txt",
"images_dir": "/home/thiagopx/data/IntegraCAR-LULC-500/satellite_images/",
"masks_dir": "/home/thiagopx/data/IntegraCAR-LULC-500/converted_masks/",
"use_patches": true,
"image_size": 2048,
"patch_size": 512,
"patch_step": 64,
"inference_stride": 64,
"architecture": "deeplabv3",
"encoder_name": "resnet50",
"encoder_weights": "imagenet",
"in_channels": 3,
"num_classes": 5,
"batch_size": 200,
"num_epochs": 100,
"learning_rate": 0.01,
"weight_decay": 0.0005,
"optimizer": "SGD",
"scheduler": "MultiStepLR",
"milestones": [
25,
35,
45
],
"gamma": 0.1,
"loss_function": "CrossEntropyLoss",
"ignore_index": 5,
"use_class_weights": true,
"experiment_name": "deeplabv3-resnet50-64-001-lr",
"use_wandb": false,
"log_every_n_batches": 0,
"num_workers": 4,
"persistent_workers": true,
"prefetch_factor": 2,
"use_amp": true,
"cudnn_benchmark": true,
"seed": 42,
"inference_overlap": 448,
"device": "cuda"
}
} |