Hugging Face's logo

RozanskiT
/
example_bundle

Model card Files
xet
 Community
example_bundle
File size: 2,231 Bytes
e02778e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
{
  "bundle": {
    "bundle_subdir": "bundle"
  },
  "hub": {
    "repo_id": null,
    "revision": null
  },
  "io": {
    "input_domain": null,
    "inputs": {
      "channel_meanings_tree": {
        "parameters": [
          "effective temperature",
          "surface gravity",
          "metallicity [Fe/H]"
        ]
      },
      "channel_names_tree": {
        "parameters": [
          "teff",
          "logg",
          "feh"
        ]
      },
      "channel_units_tree": {
        "parameters": [
          "K",
          "dex",
          "dex"
        ]
      },
      "leaf_meanings_tree": {
        "parameters": "stellar labels"
      },
      "leaf_units_tree": null,
      "structure_tree": {
        "parameters": null
      }
    },
    "outputs": {
      "channel_meanings_tree": null,
      "channel_names_tree": null,
      "channel_units_tree": null,
      "leaf_meanings_tree": {
        "flux": "continuum-normalized flux vector on the shared wavelength grid"
      },
      "leaf_units_tree": {
        "flux": "dimensionless"
      },
      "structure_tree": {
        "flux": null
      }
    },
    "reference_scaling": null
  },
  "model": {
    "init_hints": {},
    "name": "mlp",
    "params": {
      "activation": "gelu",
      "dtype": "float32",
      "hidden_sizes": [
        128,
        128
      ]
    }
  },
  "optim": {
    "b1": 0.9,
    "b2": 0.999,
    "decay_steps": 0,
    "eps": 1e-08,
    "lr": 0.001,
    "name": "soap",
    "precondition_1d": false,
    "precondition_frequency": 10,
    "schedule": "cosine",
    "warmup_steps": 1000,
    "weight_decay": 1e-05
  },
  "schema_version": 1,
  "seed": 0,
  "solver": {
    "name": "auto",
    "params": {}
  },
  "task": {
    "name": "regression",
    "params": {
      "loss": "mse"
    }
  },
  "training": {
    "batch_size": 128,
    "checkpoint_interval_steps": 500,
    "checkpoint_steps": null,
    "evaluation_interval_steps": 500,
    "evaluation_steps": null,
    "logging_interval_steps": 50,
    "logging_steps": null,
    "max_saved_checkpoints": 5,
    "num_steps": 10000,
    "shuffle": true,
    "shuffle_seed": 0,
    "steps_per_epoch": null,
    "val_fraction": 0.1,
    "workdir": "./runs/from_bundle"
  }
}