dennys246 commited on
Commit
6374609
·
verified ·
1 Parent(s): 3935449

Release v0.1.0

Browse files
MANIFEST.md ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # snowGAN release v0.1.0
2
+
3
+ Snapshot of a snowGAN training run, packaged for downstream consumers
4
+ (AvAI etc.). Consume with `snowgan.weights.fetch("RMDig/snowGAN-core", "v0.1.0")`.
5
+
6
+ ## Provenance
7
+ - snowGAN package version: `unknown`
8
+ - snowGAN git SHA at release: `9d5b8e8d9b3996e417e70df6f024795529f000b8`
9
+ - Training dataset: `rmdig/rocky_mountain_snowpack`
10
+ - Source save_dir: `/mnt/d/GitSpot/snowGAN/keras/snowgan/core`
11
+
12
+ ## Model architecture
13
+ - depth: 1
14
+ - resolution: [1024, 1024]
15
+ - modality: core
16
+ - channels: 3
17
+ - latent_dim: 100
18
+ - filter_counts (gen): [1024, 512, 256, 128, 64]
19
+ - filter_counts (disc): [64, 128, 256, 512, 1024]
20
+ - kernel_size / kernel_stride: [3, 3] / [2, 2]
21
+
22
+ ## Training state at release
23
+ - fade_step: 130000 (gen) / 130000 (disc)
24
+ - fade_steps target: 50000
25
+ - current_epoch: 0
26
+ - training_steps: gen=3, disc=2
27
+ - lambda_gp: 10.0
28
+
29
+ ## Advanced training options
30
+ - spectral_norm: False
31
+ - augment: True
32
+ - multiscale_disc: True
33
+ - ema_decay: 0.999
34
+ - lr_decay: cosine (lr_min=1e-07)
35
+ - ada_target: 0.6
36
+ - adaptive_steps: False
37
+ - grad_clip_norm: 1.0
38
+ - fid_interval: 5000
39
+
40
+ ## Persisted dataset splits
41
+ - trained_pool: 10 groups
42
+ - validation_pool: 1 groups
43
+ - test_pool: 2 groups
44
+
45
+ ## Artifacts
46
+ - `discriminator.weights.h5`
47
+ - `discriminator_config.json`
48
+ - `discriminator_lowres.weights.h5`
49
+ - `generator.weights.h5`
50
+ - `generator_config.json`
51
+ - `generator_ema.weights.h5`
52
+ - `generator_fade_endpoints.weights.h5`
53
+
54
+ ## Notes
55
+ First core release. Trained ~127k steps without spectral_norm; disc loss diverged post-step-80k due to small dataset (~13 cores) + no Lipschitz constraint + unconstrained multiscale_disc lowres critic. Generator produces structured outputs (vertical snow-like patterns, blue/white palette). Backbone Conv3D features usable for transfer; full retrain with --spectral_norm planned for v0.2.0.
discriminator.weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4d0df0399dd40631b5e2615813b5587fb3be2a74a8a6e21e38db6d3d0d11df21
3
+ size 29313976
discriminator_config.json ADDED
@@ -0,0 +1,141 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "save_dir": "keras/snowgan/core/",
3
+ "checkpoint": "keras/snowgan/core/discriminator.weights.h5",
4
+ "dataset": "rmdig/rocky_mountain_snowpack",
5
+ "datatype": "magnified_profile",
6
+ "architecture": "discriminator",
7
+ "resolution": [
8
+ 1024,
9
+ 1024
10
+ ],
11
+ "channels": 3,
12
+ "depth": 1,
13
+ "images": null,
14
+ "trained_pool": [
15
+ [
16
+ 2,
17
+ 1,
18
+ 3
19
+ ],
20
+ [
21
+ 2,
22
+ 1,
23
+ 2
24
+ ],
25
+ [
26
+ 1,
27
+ 1,
28
+ 2
29
+ ],
30
+ [
31
+ 2,
32
+ 1,
33
+ 5
34
+ ],
35
+ [
36
+ 2,
37
+ 1,
38
+ 8
39
+ ],
40
+ [
41
+ 2,
42
+ 1,
43
+ 1
44
+ ],
45
+ [
46
+ 2,
47
+ 1,
48
+ 7
49
+ ],
50
+ [
51
+ 2,
52
+ 1,
53
+ 4
54
+ ],
55
+ [
56
+ 1,
57
+ 1,
58
+ 3
59
+ ],
60
+ [
61
+ 1,
62
+ 1,
63
+ 4
64
+ ]
65
+ ],
66
+ "validation_pool": [
67
+ [
68
+ 0,
69
+ 1,
70
+ 1
71
+ ]
72
+ ],
73
+ "test_pool": [
74
+ [
75
+ 1,
76
+ 1,
77
+ 1
78
+ ],
79
+ [
80
+ 2,
81
+ 1,
82
+ 6
83
+ ]
84
+ ],
85
+ "model_history": null,
86
+ "n_samples": 10,
87
+ "epochs": 1000,
88
+ "current_epoch": 0,
89
+ "batch_size": 4,
90
+ "training_steps": 2,
91
+ "learning_rate": 1e-05,
92
+ "beta_1": 0.5,
93
+ "beta_2": 0.9,
94
+ "negative_slope": 0.25,
95
+ "lambda_gp": 10.0,
96
+ "latent_dim": 100,
97
+ "convolution_depth": 5,
98
+ "filter_counts": [
99
+ 64,
100
+ 128,
101
+ 256,
102
+ 512,
103
+ 1024
104
+ ],
105
+ "kernel_size": [
106
+ 3,
107
+ 3
108
+ ],
109
+ "kernel_stride": [
110
+ 2,
111
+ 2
112
+ ],
113
+ "batch_norm": false,
114
+ "final_activation": "tanh",
115
+ "zero_padding": null,
116
+ "padding": "same",
117
+ "optimizer": "adam",
118
+ "loss": null,
119
+ "train_ind": 0,
120
+ "trained_data": [],
121
+ "seen_profiles": [],
122
+ "rebuild": false,
123
+ "fade": false,
124
+ "fade_steps": 50000,
125
+ "fade_step": 130000,
126
+ "cleanup_milestone": 1000,
127
+ "spectral_norm": false,
128
+ "augment": true,
129
+ "lr_decay": "cosine",
130
+ "lr_min": 1e-07,
131
+ "ema_decay": 0.999,
132
+ "fid_interval": 5000,
133
+ "multiscale_disc": true,
134
+ "grad_clip_norm": 1.0,
135
+ "ada_target": 0.6,
136
+ "adaptive_steps": false,
137
+ "seed": 42,
138
+ "modality": "core",
139
+ "sample_epoch_interval": 1,
140
+ "sample_batch_interval": 10
141
+ }
discriminator_lowres.weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f2a1e6427a340c0862443e46e8b53669f809c436a16e6de3a862a1e0e3ce963
3
+ size 2559552
generator.weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:41e114e848702857b796c5a9ba13afd147d7ac324006bb0a1d031079c75deabb
3
+ size 167727864
generator_config.json ADDED
@@ -0,0 +1,141 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "save_dir": "keras/snowgan/core/",
3
+ "checkpoint": "keras/snowgan/core/generator.weights.h5",
4
+ "dataset": "rmdig/rocky_mountain_snowpack",
5
+ "datatype": "magnified_profile",
6
+ "architecture": "generator",
7
+ "resolution": [
8
+ 1024,
9
+ 1024
10
+ ],
11
+ "channels": 3,
12
+ "depth": 1,
13
+ "images": null,
14
+ "trained_pool": [
15
+ [
16
+ 2,
17
+ 1,
18
+ 3
19
+ ],
20
+ [
21
+ 2,
22
+ 1,
23
+ 2
24
+ ],
25
+ [
26
+ 1,
27
+ 1,
28
+ 2
29
+ ],
30
+ [
31
+ 2,
32
+ 1,
33
+ 5
34
+ ],
35
+ [
36
+ 2,
37
+ 1,
38
+ 8
39
+ ],
40
+ [
41
+ 2,
42
+ 1,
43
+ 1
44
+ ],
45
+ [
46
+ 2,
47
+ 1,
48
+ 7
49
+ ],
50
+ [
51
+ 2,
52
+ 1,
53
+ 4
54
+ ],
55
+ [
56
+ 1,
57
+ 1,
58
+ 3
59
+ ],
60
+ [
61
+ 1,
62
+ 1,
63
+ 4
64
+ ]
65
+ ],
66
+ "validation_pool": [
67
+ [
68
+ 0,
69
+ 1,
70
+ 1
71
+ ]
72
+ ],
73
+ "test_pool": [
74
+ [
75
+ 1,
76
+ 1,
77
+ 1
78
+ ],
79
+ [
80
+ 2,
81
+ 1,
82
+ 6
83
+ ]
84
+ ],
85
+ "model_history": null,
86
+ "n_samples": 10,
87
+ "epochs": 1000,
88
+ "current_epoch": 0,
89
+ "batch_size": 4,
90
+ "training_steps": 3,
91
+ "learning_rate": 0.0001,
92
+ "beta_1": 0.5,
93
+ "beta_2": 0.9,
94
+ "negative_slope": 0.25,
95
+ "lambda_gp": 10.0,
96
+ "latent_dim": 100,
97
+ "convolution_depth": 5,
98
+ "filter_counts": [
99
+ 1024,
100
+ 512,
101
+ 256,
102
+ 128,
103
+ 64
104
+ ],
105
+ "kernel_size": [
106
+ 3,
107
+ 3
108
+ ],
109
+ "kernel_stride": [
110
+ 2,
111
+ 2
112
+ ],
113
+ "batch_norm": false,
114
+ "final_activation": "tanh",
115
+ "zero_padding": null,
116
+ "padding": "same",
117
+ "optimizer": "adam",
118
+ "loss": null,
119
+ "train_ind": 212,
120
+ "trained_data": [],
121
+ "seen_profiles": [],
122
+ "rebuild": false,
123
+ "fade": false,
124
+ "fade_steps": 50000,
125
+ "fade_step": 130000,
126
+ "cleanup_milestone": 1000,
127
+ "spectral_norm": false,
128
+ "augment": true,
129
+ "lr_decay": "cosine",
130
+ "lr_min": 1e-07,
131
+ "ema_decay": 0.999,
132
+ "fid_interval": 5000,
133
+ "multiscale_disc": true,
134
+ "grad_clip_norm": 1.0,
135
+ "ada_target": 0.6,
136
+ "adaptive_steps": false,
137
+ "seed": 42,
138
+ "modality": "core",
139
+ "sample_epoch_interval": 1,
140
+ "sample_batch_interval": 10
141
+ }
generator_ema.weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3734686ca964afc30bac7f91b06c0c39c84d0666fb7aa9ed1a60b99b7df09cca
3
+ size 167727864
generator_fade_endpoints.weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61015391016b892ba89d47afa69082619d7221cbed844b9ca39f8ddab9d6ebaf
3
+ size 167734448