| { |
| "best_val_iou": 0.7568502187728882, |
| "final_train_loss": 0.044125475216744925, |
| "final_val_loss": 2.267013203045901, |
| "final_train_iou": 0.9337542270581554, |
| "final_val_iou": 0.7567107617855072, |
| "total_epochs": 100, |
| "patchify_enabled": true, |
| "patches_per_image": 169, |
| "inference_stride": 256, |
| "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": 128, |
| "inference_stride": 256, |
| "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-128-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": 256, |
| "device": "cuda" |
| } |
| } |
