Ziming commited on
Commit
00d5619
·
1 Parent(s): b5d3552

Add fine-tuned OlmoEarth BASE checkpoint and dataset config

Browse files
checkpoints/best.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8b16c6c16558e8f9813cbc841c09e3e4d82da65137eb6b2e363ca5be44a36aa9
3
+ size 611695455
configs/dataset_config.json ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "layers": {
3
+ "sentinel2_l2a": {
4
+ "band_sets": [{
5
+ "bands": ["B01", "B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B09", "B11", "B12"],
6
+ "dtype": "uint16"
7
+ }],
8
+ "data_source": {
9
+ "class_path": "rslearn.data_sources.planetary_computer.Sentinel2",
10
+ "init_args": {
11
+ "cache_dir": "cache/planetary_computer",
12
+ "harmonize": true,
13
+ "sort_by": "eo:cloud_cover"
14
+ },
15
+ "query_config": {
16
+ "max_matches": 4,
17
+ "period_duration": "30d",
18
+ "space_mode": "PER_PERIOD_MOSAIC"
19
+ }
20
+ },
21
+ "type": "raster"
22
+ },
23
+
24
+ "eia_energy": {
25
+ "band_sets": [{
26
+ "bands": ["label"],
27
+ "dtype": "uint8"
28
+ }],
29
+ "data_source": {
30
+ "class_path": "data_sources.eia_860.EIA860",
31
+ "init_args": {
32
+ "csv_path": "source_data/eia860/3_1_Generator_Y2024_merged.csv",
33
+ "buffer_meters": 500,
34
+ "min_capacity_mw": 1.0
35
+ }
36
+ },
37
+ "resampling_method": "nearest",
38
+ "type": "raster"
39
+ },
40
+
41
+ "output": {
42
+ "band_sets": [{
43
+ "bands": ["output"],
44
+ "dtype": "uint8"
45
+ }],
46
+ "type": "raster"
47
+ }
48
+ }
49
+ }