Commit
•
eae0e6c
1
Parent(s):
647e7a5
Push Keras model using huggingface_hub.
Browse files- .gitattributes +1 -0
- README.md +54 -0
- config.json +1 -0
- fingerprint.pb +3 -0
- keras_metadata.pb +3 -0
- model.png +0 -0
- saved_model.pb +3 -0
- variables/variables.data-00000-of-00001 +3 -0
- variables/variables.index +0 -0
.gitattributes
CHANGED
@@ -32,3 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
32 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
33 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
34 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
32 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
33 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
34 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
35 |
+
variables/variables.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
@@ -0,0 +1,54 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
library_name: keras
|
3 |
+
---
|
4 |
+
|
5 |
+
## Model description
|
6 |
+
|
7 |
+
More information needed
|
8 |
+
|
9 |
+
## Intended uses & limitations
|
10 |
+
|
11 |
+
More information needed
|
12 |
+
|
13 |
+
## Training and evaluation data
|
14 |
+
|
15 |
+
More information needed
|
16 |
+
|
17 |
+
## Training procedure
|
18 |
+
|
19 |
+
### Training hyperparameters
|
20 |
+
|
21 |
+
The following hyperparameters were used during training:
|
22 |
+
|
23 |
+
| Hyperparameters | Value |
|
24 |
+
| :-- | :-- |
|
25 |
+
| inner_optimizer.class_name | Custom>RMSprop |
|
26 |
+
| inner_optimizer.config.name | RMSprop |
|
27 |
+
| inner_optimizer.config.weight_decay | None |
|
28 |
+
| inner_optimizer.config.clipnorm | None |
|
29 |
+
| inner_optimizer.config.global_clipnorm | None |
|
30 |
+
| inner_optimizer.config.clipvalue | None |
|
31 |
+
| inner_optimizer.config.use_ema | False |
|
32 |
+
| inner_optimizer.config.ema_momentum | 0.99 |
|
33 |
+
| inner_optimizer.config.ema_overwrite_frequency | 100 |
|
34 |
+
| inner_optimizer.config.jit_compile | True |
|
35 |
+
| inner_optimizer.config.is_legacy_optimizer | False |
|
36 |
+
| inner_optimizer.config.learning_rate | 0.0010000000474974513 |
|
37 |
+
| inner_optimizer.config.rho | 0.9 |
|
38 |
+
| inner_optimizer.config.momentum | 0.0 |
|
39 |
+
| inner_optimizer.config.epsilon | 1e-07 |
|
40 |
+
| inner_optimizer.config.centered | False |
|
41 |
+
| dynamic | True |
|
42 |
+
| initial_scale | 32768.0 |
|
43 |
+
| dynamic_growth_steps | 2000 |
|
44 |
+
| training_precision | mixed_float16 |
|
45 |
+
|
46 |
+
|
47 |
+
## Model Plot
|
48 |
+
|
49 |
+
<details>
|
50 |
+
<summary>View Model Plot</summary>
|
51 |
+
|
52 |
+
![Model Image](./model.png)
|
53 |
+
|
54 |
+
</details>
|
config.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
{"name": "diffusion_model_2", "layers": [{"class_name": "InputLayer", "config": {"batch_input_shape": [null, 320], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_10"}, "name": "input_10", "inbound_nodes": []}, {"class_name": "Dense", "config": {"name": "dense_512", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "units": 1280, "activation": "linear", "use_bias": true, "kernel_initializer": {"class_name": "GlorotUniform", "config": {"seed": null}}, "bias_initializer": {"class_name": "Zeros", "config": {}}, "kernel_regularizer": null, "bias_regularizer": null, "activity_regularizer": null, "kernel_constraint": null, "bias_constraint": null}, "name": "dense_512", "inbound_nodes": [[["input_10", 0, 0, {}]]]}, {"class_name": "InputLayer", "config": {"batch_input_shape": [null, 64, 64, 4], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_11"}, "name": "input_11", "inbound_nodes": []}, {"class_name": "Activation", "config": {"name": "activation_138", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "activation": "swish"}, "name": "activation_138", "inbound_nodes": [[["dense_512", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_236", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "kernel_size": 3, "padding": 1, "filters": 320}, "name": "padded_conv2d_236", "inbound_nodes": [[["input_11", 0, 0, {}]]]}, {"class_name": "Dense", "config": {"name": "dense_513", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "units": 1280, "activation": "linear", "use_bias": true, "kernel_initializer": {"class_name": "GlorotUniform", "config": {"seed": null}}, "bias_initializer": {"class_name": "Zeros", "config": {}}, "kernel_regularizer": null, "bias_regularizer": null, "activity_regularizer": null, "kernel_constraint": null, "bias_constraint": null}, "name": "dense_513", "inbound_nodes": [[["activation_138", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_44", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_44", "inbound_nodes": [[["padded_conv2d_236", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "InputLayer", "config": {"batch_input_shape": [null, 77, 768], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_9"}, "name": "input_9", "inbound_nodes": []}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_32", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "fully_connected": false, "num_heads": 8, "head_size": 40}, "name": "spatial_transformer_32", "inbound_nodes": [[["res_block_44", 0, 0, {}], ["input_9", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_45", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_45", "inbound_nodes": [[["spatial_transformer_32", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_33", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "fully_connected": false, "num_heads": 8, "head_size": 40}, "name": "spatial_transformer_33", "inbound_nodes": [[["res_block_45", 0, 0, {}], ["input_9", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_245", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "strides": 2, "padding": 1, "filters": 320, "kernel_size": 3}, "name": "padded_conv2d_245", "inbound_nodes": [[["spatial_transformer_33", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_46", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_46", "inbound_nodes": [[["padded_conv2d_245", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_34", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "fully_connected": false, "num_heads": 8, "head_size": 80}, "name": "spatial_transformer_34", "inbound_nodes": [[["res_block_46", 0, 0, {}], ["input_9", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_47", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_47", "inbound_nodes": [[["spatial_transformer_34", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_35", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "fully_connected": false, "num_heads": 8, "head_size": 80}, "name": "spatial_transformer_35", "inbound_nodes": [[["res_block_47", 0, 0, {}], ["input_9", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_254", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "strides": 2, "padding": 1, "filters": 640, "kernel_size": 3}, "name": "padded_conv2d_254", "inbound_nodes": [[["spatial_transformer_35", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_48", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_48", "inbound_nodes": [[["padded_conv2d_254", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_36", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "fully_connected": false, "num_heads": 8, "head_size": 160}, "name": "spatial_transformer_36", "inbound_nodes": [[["res_block_48", 0, 0, {}], ["input_9", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_49", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_49", "inbound_nodes": [[["spatial_transformer_36", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_37", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "fully_connected": false, "num_heads": 8, "head_size": 160}, "name": "spatial_transformer_37", "inbound_nodes": [[["res_block_49", 0, 0, {}], ["input_9", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_263", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "strides": 2, "padding": 1, "filters": 1280, "kernel_size": 3}, "name": "padded_conv2d_263", "inbound_nodes": [[["spatial_transformer_37", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_50", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_50", "inbound_nodes": [[["padded_conv2d_263", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_51", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_51", "inbound_nodes": [[["res_block_50", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_52", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_52", "inbound_nodes": [[["res_block_51", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_38", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "fully_connected": false, "num_heads": 8, "head_size": 160}, "name": "spatial_transformer_38", "inbound_nodes": [[["res_block_52", 0, 0, {}], ["input_9", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_53", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_53", "inbound_nodes": [[["spatial_transformer_38", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_24", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_24", "inbound_nodes": [[["res_block_53", 0, 0, {}], ["res_block_51", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_54", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_54", "inbound_nodes": [[["concatenate_24", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_25", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_25", "inbound_nodes": [[["res_block_54", 0, 0, {}], ["res_block_50", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_55", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_55", "inbound_nodes": [[["concatenate_25", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_26", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_26", "inbound_nodes": [[["res_block_55", 0, 0, {}], ["padded_conv2d_263", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_56", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_56", "inbound_nodes": [[["concatenate_26", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "Upsample", "config": {"name": "upsample_6", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "channels": 1280}, "name": "upsample_6", "inbound_nodes": [[["res_block_56", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_27", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_27", "inbound_nodes": [[["upsample_6", 0, 0, {}], ["spatial_transformer_37", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_57", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_57", "inbound_nodes": [[["concatenate_27", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_39", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "fully_connected": false, "num_heads": 8, "head_size": 160}, "name": "spatial_transformer_39", "inbound_nodes": [[["res_block_57", 0, 0, {}], ["input_9", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_28", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_28", "inbound_nodes": [[["spatial_transformer_39", 0, 0, {}], ["spatial_transformer_36", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_58", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_58", "inbound_nodes": [[["concatenate_28", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_40", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "fully_connected": false, "num_heads": 8, "head_size": 160}, "name": "spatial_transformer_40", "inbound_nodes": [[["res_block_58", 0, 0, {}], ["input_9", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_29", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_29", "inbound_nodes": [[["spatial_transformer_40", 0, 0, {}], ["padded_conv2d_254", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_59", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_59", "inbound_nodes": [[["concatenate_29", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_41", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "fully_connected": false, "num_heads": 8, "head_size": 160}, "name": "spatial_transformer_41", "inbound_nodes": [[["res_block_59", 0, 0, {}], ["input_9", 0, 0, {}]]]}, {"class_name": "Upsample", "config": {"name": "upsample_7", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "channels": 1280}, "name": "upsample_7", "inbound_nodes": [[["spatial_transformer_41", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_30", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_30", "inbound_nodes": [[["upsample_7", 0, 0, {}], ["spatial_transformer_35", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_60", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_60", "inbound_nodes": [[["concatenate_30", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_42", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "fully_connected": false, "num_heads": 8, "head_size": 80}, "name": "spatial_transformer_42", "inbound_nodes": [[["res_block_60", 0, 0, {}], ["input_9", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_31", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_31", "inbound_nodes": [[["spatial_transformer_42", 0, 0, {}], ["spatial_transformer_34", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_61", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_61", "inbound_nodes": [[["concatenate_31", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_43", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "fully_connected": false, "num_heads": 8, "head_size": 80}, "name": "spatial_transformer_43", "inbound_nodes": [[["res_block_61", 0, 0, {}], ["input_9", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_32", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_32", "inbound_nodes": [[["spatial_transformer_43", 0, 0, {}], ["padded_conv2d_245", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_62", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_62", "inbound_nodes": [[["concatenate_32", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_44", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "fully_connected": false, "num_heads": 8, "head_size": 80}, "name": "spatial_transformer_44", "inbound_nodes": [[["res_block_62", 0, 0, {}], ["input_9", 0, 0, {}]]]}, {"class_name": "Upsample", "config": {"name": "upsample_8", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "channels": 640}, "name": "upsample_8", "inbound_nodes": [[["spatial_transformer_44", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_33", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_33", "inbound_nodes": [[["upsample_8", 0, 0, {}], ["spatial_transformer_33", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_63", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_63", "inbound_nodes": [[["concatenate_33", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_45", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "fully_connected": false, "num_heads": 8, "head_size": 40}, "name": "spatial_transformer_45", "inbound_nodes": [[["res_block_63", 0, 0, {}], ["input_9", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_34", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_34", "inbound_nodes": [[["spatial_transformer_45", 0, 0, {}], ["spatial_transformer_32", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_64", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_64", "inbound_nodes": [[["concatenate_34", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_46", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "fully_connected": false, "num_heads": 8, "head_size": 40}, "name": "spatial_transformer_46", "inbound_nodes": [[["res_block_64", 0, 0, {}], ["input_9", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_35", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_35", "inbound_nodes": [[["spatial_transformer_46", 0, 0, {}], ["padded_conv2d_236", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_65", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_65", "inbound_nodes": [[["concatenate_35", 0, 0, {}], ["dense_513", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_47", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "fully_connected": false, "num_heads": 8, "head_size": 40}, "name": "spatial_transformer_47", "inbound_nodes": [[["res_block_65", 0, 0, {}], ["input_9", 0, 0, {}]]]}, {"class_name": "GroupNormalization", "config": {"name": "group_normalization_234", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "groups": 32, "axis": -1, "epsilon": 1e-05, "center": true, "scale": true, "beta_initializer": {"class_name": "Zeros", "config": {}}, "gamma_initializer": {"class_name": "Ones", "config": {}}, "beta_regularizer": null, "gamma_regularizer": null, "beta_constraint": null, "gamma_constraint": null}, "name": "group_normalization_234", "inbound_nodes": [[["spatial_transformer_47", 0, 0, {}]]]}, {"class_name": "Activation", "config": {"name": "activation_205", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "activation": "swish"}, "name": "activation_205", "inbound_nodes": [[["group_normalization_234", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_319", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "kernel_size": 3, "padding": 1, "filters": 4}, "name": "padded_conv2d_319", "inbound_nodes": [[["activation_205", 0, 0, {}]]]}], "input_layers": [["input_11", 0, 0], ["input_10", 0, 0], ["input_9", 0, 0]], "output_layers": [["padded_conv2d_319", 0, 0]]}
|
fingerprint.pb
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0655551d93780b6fca2d3ca67971ab99c01db8000e8833cc639a0aec47dd3f7b
|
3 |
+
size 55
|
keras_metadata.pb
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:140ba2f8b30f9ea94e838f6260e9b9df9c5eb3de65fbe29ec89ffa6d5146c140
|
3 |
+
size 861075
|
model.png
ADDED
![]() |
saved_model.pb
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6c661cbe62568ad19d15f37096fd13fd1a29f69a0873d78735e606a7ce438a07
|
3 |
+
size 20839230
|
variables/variables.data-00000-of-00001
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a4ca6e2b9913be869a6c714e06c79f529ccc8e4523a4d868bb53cc20dd9e7367
|
3 |
+
size 3438636316
|
variables/variables.index
ADDED
Binary file (40.5 kB). View file
|
|