yx21e commited on
Commit
20ce74a
·
verified ·
1 Parent(s): 3907f52

Clarify California scope and add CONUS retraining recipe

Browse files
training/configs/train_firewx_fm_conus_seed7_template.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "index_root": "/path/to/local/firewx_fm_cache/conus_hrrr_us_5km_l12",
3
+ "run_root": "/path/to/local/firewx_fm_runs",
4
+ "tile_size": 32,
5
+ "max_positive_tiles_per_sample": 96,
6
+ "min_negative_tiles_per_sample": 8,
7
+ "negative_to_positive_ratio": 3.0,
8
+ "positive_rate_source": "tiles",
9
+ "pos_weight_source": "tiles",
10
+ "pos_weight_cap": 50.0,
11
+ "batch_size": 32,
12
+ "eval_batch_size": 1,
13
+ "epochs": 10,
14
+ "learning_rate": 0.0002,
15
+ "weight_decay": 0.0001,
16
+ "base_channels": 32,
17
+ "dropout": 0.1,
18
+ "norm_type": "group",
19
+ "norm_groups": 8,
20
+ "init_head_bias_from_positive_rate": true,
21
+ "augment_flip": true,
22
+ "threshold": 0.5,
23
+ "metric_thresholds": [
24
+ 0.1,
25
+ 0.2,
26
+ 0.3,
27
+ 0.5,
28
+ 0.7,
29
+ 0.9
30
+ ],
31
+ "topk_area_fractions": [
32
+ 0.01,
33
+ 0.05,
34
+ 0.1
35
+ ],
36
+ "fss_radii": [
37
+ 1,
38
+ 2,
39
+ 4,
40
+ 8
41
+ ],
42
+ "reliability_bins": 10,
43
+ "amp": true,
44
+ "seed": 7,
45
+ "use_aux_spatial_head": true,
46
+ "aux_spatial_radius": 2,
47
+ "aux_spatial_loss_weight": 0.5,
48
+ "train_target_mode": "dilate_max",
49
+ "train_target_radius": 2,
50
+ "loss_type": "bce",
51
+ "init_checkpoint": ""
52
+ }