Upload 2 files

congifs/dataloader_full.toml

@@ -0,0 +1,16 @@
+[dataset]
+crop_manifest_path = "data/crops/crop_manifest.csv"
+include_alpha = true
+
+[dataloader]
+batch_size = 2
+shuffle = true
+num_workers = 4
+pin_memory = true
+
+[normalization]
+scale = 10000.0
+clip_min = 0.0
+clip_max = 4.0
+band_mean = [0.0610578594098199, 0.07939659505065455, 0.3141591574037171, 0.6759725915130503]
+band_std = [0.05220907362214255, 0.05130135127966304, 0.20160384174129178, 0.40058156788879246]

congifs/train_mitb2_full.toml

@@ -0,0 +1,52 @@
+[training]
+data_config = "configs/dataloader_full.toml"
+output_path = "outputs/train_mitb2_full_summary.json"
+checkpoint_path = "outputs/checkpoints/mitb2_full.ckpt"
+resume_from_checkpoint = ""
+validation_fraction = 0.1
+seed = 13
+
+[trainer]
+max_steps = 100000
+max_validation_batches = 64
+accelerator = "gpu"
+devices = 1
+precision = "16-mixed"
+early_stopping = true
+early_stopping_patience = 20
+checkpoint_monitor = "val/loss"
+log_every_n_steps = 25
+val_check_interval = 500
+num_sanity_val_steps = 1
+gradient_clip_val = 1.0
+enable_progress_bar = true
+
+[logging]
+csv_logging = true
+log_dir = "outputs/logs"
+run_name = "mitb2_full"
+
+[optimizer]
+learning_rate = 0.00005
+weight_decay = 0.0001
+
+[objective]
+contrastive_weight = 1.0
+reconstruction_weight = 1.0
+mask_probability = 0.3
+contrastive_temperature = 0.2
+
+[model]
+in_channels = 4
+spatial_backbone = "segformer_mit_b2"
+spatial_dim = 256
+temporal_dim = 256
+embedding_dim = 256
+transformer_layers = 2
+transformer_heads = 4
+dropout = 0.1
+segformer_model_name = "nvidia/mit-b2"
+segformer_pretrained = true
+segformer_local_files_only = false
+freeze_spatial_encoder = true
+segformer_input_size = 224