wesselvannierop commited on
Commit
74322bb
·
verified ·
1 Parent(s): 6bee7f6

Add 3 frame ULSA model

Browse files
Files changed (2) hide show
  1. config.json +54 -0
  2. model.weights.h5 +3 -0
config.json ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "module": "zea.models.diffusion",
3
+ "class_name": "DiffusionModel",
4
+ "config": {
5
+ "name": "diffusion",
6
+ "trainable": true,
7
+ "dtype": {
8
+ "module": "keras",
9
+ "class_name": "DTypePolicy",
10
+ "config": {
11
+ "name": "float32"
12
+ },
13
+ "registered_name": null
14
+ },
15
+ "input_shape": [
16
+ 112,
17
+ 112,
18
+ 3
19
+ ],
20
+ "network_name": "unet_time_conditional",
21
+ "network_kwargs": {
22
+ "block_depth": 2,
23
+ "widths": [
24
+ 32,
25
+ 64,
26
+ 96,
27
+ 128
28
+ ]
29
+ },
30
+ "input_range": [
31
+ -1,
32
+ 1
33
+ ],
34
+ "min_signal_rate": 0.02,
35
+ "max_signal_rate": 0.95
36
+ },
37
+ "registered_name": "DiffusionModel",
38
+ "build_config": {
39
+ "input_shape": [
40
+ [
41
+ 1,
42
+ 112,
43
+ 112,
44
+ 3
45
+ ],
46
+ [
47
+ 1,
48
+ 1,
49
+ 1,
50
+ 1
51
+ ]
52
+ ]
53
+ }
54
+ }
model.weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e9d7193a5aa8efaf0e59d4ce5e118d05147ee95a077dcdabb49d1ae77ed8bd2
3
+ size 31731280