kylecompute commited on
Commit
4ecb57a
·
verified ·
1 Parent(s): a1f94cd

model card: correct weight paths

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -31,8 +31,8 @@ The polar sun circles the horizon, so a shadow cast from thirty directions is a
31
 
32
  | path | contents |
33
  |---|---|
34
- | `weights/<site>/<run>/net.pt` | UNet state dict, PyTorch. 198 checkpoints: 7 to 8 seeds per 6 km site (`unet1` …), 3 per 16 km site, one `long` run, 20 MB each |
35
- | `weights/<site>/<run>/params.json` | the per-sun-bin exposure and floor fitted with that network |
36
  | `weights/<site>/split.json` | the frozen train/test split of sun bins for that site |
37
  | `weights/<site>/sun_fit.csv` | sun direction per bin, nominal and refined |
38
  | `results/<site>.csv` | held-out scores for every map at that site, plain column names |
@@ -67,7 +67,7 @@ Reproduce a site's map from the public inputs with the pipeline in the code repo
67
  export MOONDEM_DATA=data/sites/haworth MOONDEM_BOX=haworth
68
  python pipeline/step1_cut_photos.py && python pipeline/step2_get_laser_heights.py && python pipeline/step3_check_sun_angles.py
69
  cp weights/haworth/split.json $MOONDEM_DATA/
70
- python pipeline/step5_fit_terrain_network.py --infer-only weights/haworth/unet1/net.pt --tag from_checkpoint
71
  ```
72
 
73
  The architecture is a five-level UNet (widths 32, 64, 128, 256, 256) that takes 2·N+1 channels (N training bins, their valid masks, the LOLA base) and outputs a height residual scaled by 5 m. Input stacks, LOLA windows and the solver are in the code repository.
 
31
 
32
  | path | contents |
33
  |---|---|
34
+ | `weights/<site>/unet/<run>/net.pt` | UNet state dict, PyTorch. 198 checkpoints: 7 to 8 seeds per 6 km site (`unet1` …), 3 per 16 km site, one `long` run, 20 MB each |
35
+ | `weights/<site>/unet/<run>/params.json` | the per-sun-bin exposure and floor fitted with that network |
36
  | `weights/<site>/split.json` | the frozen train/test split of sun bins for that site |
37
  | `weights/<site>/sun_fit.csv` | sun direction per bin, nominal and refined |
38
  | `results/<site>.csv` | held-out scores for every map at that site, plain column names |
 
67
  export MOONDEM_DATA=data/sites/haworth MOONDEM_BOX=haworth
68
  python pipeline/step1_cut_photos.py && python pipeline/step2_get_laser_heights.py && python pipeline/step3_check_sun_angles.py
69
  cp weights/haworth/split.json $MOONDEM_DATA/
70
+ python pipeline/step5_fit_terrain_network.py --infer-only weights/haworth/unet/unet1/net.pt --tag from_checkpoint
71
  ```
72
 
73
  The architecture is a five-level UNet (widths 32, 64, 128, 256, 256) that takes 2·N+1 channels (N training bins, their valid masks, the LOLA base) and outputs a height residual scaled by 5 m. Input stacks, LOLA windows and the solver are in the code repository.