Push Keras model using huggingface_hub.

README.md

@@ -1,27 +1,18 @@
 ---
 library_name: keras
-license: apache-2.0
-tags:
-- keras
-- stable-diffusion
-- text-to-image
-- keras-dreambooth
-- consentful
 ---
 
 ## Model description
 
-This is a Keras Dreambooth model fine-tuned to some 18th century etching artworks by italian artist [Giambattista Piranesi](https://en.wikipedia.org/wiki/Giovanni_Battista_Piranesi).
+More information needed
 
-A few samples below
+## Intended uses & limitations
 
-![](output-samples/db-piranesi_0.jpg) ![](output-samples/db-piranesi_1.jpg) ![](output-samples/db-piranesi_2.jpg) ![](output-samples/db-piranesi_3.jpg)
-![](output-samples/db-piranesi_4.jpg)
+More information needed
 
-## Intended uses & limitations
+## Training and evaluation data
 
-The base prompt is "image of monuments in sks style". 
-A slightly better prompt would be "image of monuments in sks style, 8k, high quality, old paper" with negative prompt "distorted".
+More information needed
 
 ## Training procedure
 

config.json

@@ -1 +1 @@
-{"name": "diffusion_model_1", "layers": [{"class_name": "InputLayer", "config": {"batch_input_shape": [null, 320], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_6"}, "name": "input_6", "inbound_nodes": []}, {"class_name": "Dense", "config": {"name": "dense_256", "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_256", "inbound_nodes": [[["input_6", 0, 0, {}]]]}, {"class_name": "InputLayer", "config": {"batch_input_shape": [null, 32, 32, 4], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_7"}, "name": "input_7", "inbound_nodes": []}, {"class_name": "Activation", "config": {"name": "activation_69", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "activation": "swish"}, "name": "activation_69", "inbound_nodes": [[["dense_256", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_114", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "kernel_size": 3, "padding": 1, "filters": 320}, "name": "padded_conv2d_114", "inbound_nodes": [[["input_7", 0, 0, {}]]]}, {"class_name": "Dense", "config": {"name": "dense_257", "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_257", "inbound_nodes": [[["activation_69", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_22", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_22", "inbound_nodes": [[["padded_conv2d_114", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "InputLayer", "config": {"batch_input_shape": [null, 77, 768], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_5"}, "name": "input_5", "inbound_nodes": []}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_16", "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_16", "inbound_nodes": [[["res_block_22", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_23", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_23", "inbound_nodes": [[["spatial_transformer_16", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_17", "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_17", "inbound_nodes": [[["res_block_23", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_123", "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_123", "inbound_nodes": [[["spatial_transformer_17", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_24", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_24", "inbound_nodes": [[["padded_conv2d_123", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_18", "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_18", "inbound_nodes": [[["res_block_24", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_25", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_25", "inbound_nodes": [[["spatial_transformer_18", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_19", "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_19", "inbound_nodes": [[["res_block_25", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_132", "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_132", "inbound_nodes": [[["spatial_transformer_19", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_26", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_26", "inbound_nodes": [[["padded_conv2d_132", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_20", "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_20", "inbound_nodes": [[["res_block_26", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_27", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_27", "inbound_nodes": [[["spatial_transformer_20", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_21", "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_21", "inbound_nodes": [[["res_block_27", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_141", "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_141", "inbound_nodes": [[["spatial_transformer_21", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_28", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_28", "inbound_nodes": [[["padded_conv2d_141", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_29", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_29", "inbound_nodes": [[["res_block_28", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_30", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_30", "inbound_nodes": [[["res_block_29", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_22", "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_22", "inbound_nodes": [[["res_block_30", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_31", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_31", "inbound_nodes": [[["spatial_transformer_22", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_12", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_12", "inbound_nodes": [[["res_block_31", 0, 0, {}], ["res_block_29", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_32", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_32", "inbound_nodes": [[["concatenate_12", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_13", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_13", "inbound_nodes": [[["res_block_32", 0, 0, {}], ["res_block_28", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_33", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_33", "inbound_nodes": [[["concatenate_13", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_14", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_14", "inbound_nodes": [[["res_block_33", 0, 0, {}], ["padded_conv2d_141", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_34", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_34", "inbound_nodes": [[["concatenate_14", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "Upsample", "config": {"name": "upsample_3", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "channels": 1280}, "name": "upsample_3", "inbound_nodes": [[["res_block_34", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_15", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_15", "inbound_nodes": [[["upsample_3", 0, 0, {}], ["spatial_transformer_21", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_35", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_35", "inbound_nodes": [[["concatenate_15", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_23", "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_23", "inbound_nodes": [[["res_block_35", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_16", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_16", "inbound_nodes": [[["spatial_transformer_23", 0, 0, {}], ["spatial_transformer_20", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_36", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_36", "inbound_nodes": [[["concatenate_16", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_24", "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_24", "inbound_nodes": [[["res_block_36", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_17", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_17", "inbound_nodes": [[["spatial_transformer_24", 0, 0, {}], ["padded_conv2d_132", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_37", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_37", "inbound_nodes": [[["concatenate_17", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_25", "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_25", "inbound_nodes": [[["res_block_37", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "Upsample", "config": {"name": "upsample_4", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "channels": 1280}, "name": "upsample_4", "inbound_nodes": [[["spatial_transformer_25", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_18", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_18", "inbound_nodes": [[["upsample_4", 0, 0, {}], ["spatial_transformer_19", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_38", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_38", "inbound_nodes": [[["concatenate_18", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_26", "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_26", "inbound_nodes": [[["res_block_38", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_19", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_19", "inbound_nodes": [[["spatial_transformer_26", 0, 0, {}], ["spatial_transformer_18", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_39", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_39", "inbound_nodes": [[["concatenate_19", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_27", "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_27", "inbound_nodes": [[["res_block_39", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_20", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_20", "inbound_nodes": [[["spatial_transformer_27", 0, 0, {}], ["padded_conv2d_123", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_40", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_40", "inbound_nodes": [[["concatenate_20", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_28", "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_28", "inbound_nodes": [[["res_block_40", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "Upsample", "config": {"name": "upsample_5", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "channels": 640}, "name": "upsample_5", "inbound_nodes": [[["spatial_transformer_28", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_21", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_21", "inbound_nodes": [[["upsample_5", 0, 0, {}], ["spatial_transformer_17", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_41", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_41", "inbound_nodes": [[["concatenate_21", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_29", "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_29", "inbound_nodes": [[["res_block_41", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_22", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_22", "inbound_nodes": [[["spatial_transformer_29", 0, 0, {}], ["spatial_transformer_16", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_42", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_42", "inbound_nodes": [[["concatenate_22", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_30", "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_30", "inbound_nodes": [[["res_block_42", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_23", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_23", "inbound_nodes": [[["spatial_transformer_30", 0, 0, {}], ["padded_conv2d_114", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_43", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_43", "inbound_nodes": [[["concatenate_23", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_31", "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_31", "inbound_nodes": [[["res_block_43", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "GroupNormalization", "config": {"name": "group_normalization_143", "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_143", "inbound_nodes": [[["spatial_transformer_31", 0, 0, {}]]]}, {"class_name": "Activation", "config": {"name": "activation_136", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "activation": "swish"}, "name": "activation_136", "inbound_nodes": [[["group_normalization_143", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_197", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "kernel_size": 3, "padding": 1, "filters": 4}, "name": "padded_conv2d_197", "inbound_nodes": [[["activation_136", 0, 0, {}]]]}], "input_layers": [["input_7", 0, 0], ["input_6", 0, 0], ["input_5", 0, 0]], "output_layers": [["padded_conv2d_197", 0, 0]]}
+{"name": "diffusion_model_1", "layers": [{"class_name": "InputLayer", "config": {"batch_input_shape": [null, 320], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_6"}, "name": "input_6", "inbound_nodes": []}, {"class_name": "Dense", "config": {"name": "dense_256", "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_256", "inbound_nodes": [[["input_6", 0, 0, {}]]]}, {"class_name": "InputLayer", "config": {"batch_input_shape": [null, 64, 64, 4], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_7"}, "name": "input_7", "inbound_nodes": []}, {"class_name": "Activation", "config": {"name": "activation_69", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "activation": "swish"}, "name": "activation_69", "inbound_nodes": [[["dense_256", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_114", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "kernel_size": 3, "padding": 1, "filters": 320}, "name": "padded_conv2d_114", "inbound_nodes": [[["input_7", 0, 0, {}]]]}, {"class_name": "Dense", "config": {"name": "dense_257", "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_257", "inbound_nodes": [[["activation_69", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_22", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_22", "inbound_nodes": [[["padded_conv2d_114", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "InputLayer", "config": {"batch_input_shape": [null, 77, 768], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_5"}, "name": "input_5", "inbound_nodes": []}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_16", "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_16", "inbound_nodes": [[["res_block_22", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_23", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_23", "inbound_nodes": [[["spatial_transformer_16", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_17", "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_17", "inbound_nodes": [[["res_block_23", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_123", "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_123", "inbound_nodes": [[["spatial_transformer_17", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_24", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_24", "inbound_nodes": [[["padded_conv2d_123", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_18", "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_18", "inbound_nodes": [[["res_block_24", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_25", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_25", "inbound_nodes": [[["spatial_transformer_18", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_19", "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_19", "inbound_nodes": [[["res_block_25", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_132", "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_132", "inbound_nodes": [[["spatial_transformer_19", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_26", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_26", "inbound_nodes": [[["padded_conv2d_132", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_20", "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_20", "inbound_nodes": [[["res_block_26", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_27", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_27", "inbound_nodes": [[["spatial_transformer_20", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_21", "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_21", "inbound_nodes": [[["res_block_27", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_141", "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_141", "inbound_nodes": [[["spatial_transformer_21", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_28", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_28", "inbound_nodes": [[["padded_conv2d_141", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_29", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_29", "inbound_nodes": [[["res_block_28", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_30", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_30", "inbound_nodes": [[["res_block_29", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_22", "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_22", "inbound_nodes": [[["res_block_30", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_31", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_31", "inbound_nodes": [[["spatial_transformer_22", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_12", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_12", "inbound_nodes": [[["res_block_31", 0, 0, {}], ["res_block_29", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_32", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_32", "inbound_nodes": [[["concatenate_12", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_13", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_13", "inbound_nodes": [[["res_block_32", 0, 0, {}], ["res_block_28", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_33", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_33", "inbound_nodes": [[["concatenate_13", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_14", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_14", "inbound_nodes": [[["res_block_33", 0, 0, {}], ["padded_conv2d_141", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_34", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_34", "inbound_nodes": [[["concatenate_14", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "Upsample", "config": {"name": "upsample_3", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "channels": 1280}, "name": "upsample_3", "inbound_nodes": [[["res_block_34", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_15", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_15", "inbound_nodes": [[["upsample_3", 0, 0, {}], ["spatial_transformer_21", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_35", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_35", "inbound_nodes": [[["concatenate_15", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_23", "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_23", "inbound_nodes": [[["res_block_35", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_16", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_16", "inbound_nodes": [[["spatial_transformer_23", 0, 0, {}], ["spatial_transformer_20", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_36", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_36", "inbound_nodes": [[["concatenate_16", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_24", "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_24", "inbound_nodes": [[["res_block_36", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_17", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_17", "inbound_nodes": [[["spatial_transformer_24", 0, 0, {}], ["padded_conv2d_132", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_37", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_37", "inbound_nodes": [[["concatenate_17", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_25", "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_25", "inbound_nodes": [[["res_block_37", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "Upsample", "config": {"name": "upsample_4", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "channels": 1280}, "name": "upsample_4", "inbound_nodes": [[["spatial_transformer_25", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_18", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_18", "inbound_nodes": [[["upsample_4", 0, 0, {}], ["spatial_transformer_19", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_38", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_38", "inbound_nodes": [[["concatenate_18", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_26", "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_26", "inbound_nodes": [[["res_block_38", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_19", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_19", "inbound_nodes": [[["spatial_transformer_26", 0, 0, {}], ["spatial_transformer_18", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_39", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_39", "inbound_nodes": [[["concatenate_19", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_27", "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_27", "inbound_nodes": [[["res_block_39", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_20", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_20", "inbound_nodes": [[["spatial_transformer_27", 0, 0, {}], ["padded_conv2d_123", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_40", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_40", "inbound_nodes": [[["concatenate_20", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_28", "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_28", "inbound_nodes": [[["res_block_40", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "Upsample", "config": {"name": "upsample_5", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "channels": 640}, "name": "upsample_5", "inbound_nodes": [[["spatial_transformer_28", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_21", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_21", "inbound_nodes": [[["upsample_5", 0, 0, {}], ["spatial_transformer_17", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_41", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_41", "inbound_nodes": [[["concatenate_21", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_29", "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_29", "inbound_nodes": [[["res_block_41", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_22", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_22", "inbound_nodes": [[["spatial_transformer_29", 0, 0, {}], ["spatial_transformer_16", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_42", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_42", "inbound_nodes": [[["concatenate_22", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_30", "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_30", "inbound_nodes": [[["res_block_42", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_23", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_23", "inbound_nodes": [[["spatial_transformer_30", 0, 0, {}], ["padded_conv2d_114", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_43", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_43", "inbound_nodes": [[["concatenate_23", 0, 0, {}], ["dense_257", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_31", "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_31", "inbound_nodes": [[["res_block_43", 0, 0, {}], ["input_5", 0, 0, {}]]]}, {"class_name": "GroupNormalization", "config": {"name": "group_normalization_143", "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_143", "inbound_nodes": [[["spatial_transformer_31", 0, 0, {}]]]}, {"class_name": "Activation", "config": {"name": "activation_136", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "activation": "swish"}, "name": "activation_136", "inbound_nodes": [[["group_normalization_143", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_197", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "kernel_size": 3, "padding": 1, "filters": 4}, "name": "padded_conv2d_197", "inbound_nodes": [[["activation_136", 0, 0, {}]]]}], "input_layers": [["input_7", 0, 0], ["input_6", 0, 0], ["input_5", 0, 0]], "output_layers": [["padded_conv2d_197", 0, 0]]}

fingerprint.pb

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9858fd81e229fc549fc0363bd904ece2e8bc282b789c8e2466406bb14c754973
+oid sha256:365cb94765b36960363c7036e46faf7ae280791a75c52c99976f58cf6332ea04
 size 55

keras_metadata.pb

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c71d78911f98bd02c268ec8d7c7c2202570f7121812c134bdd479835665d7d01
-size 860492
+oid sha256:95f3ff9e20346bf9a37b1e6d42df546535d30b4a713246418b6c72b5b07b3d47
+size 860888

saved_model.pb

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6ceb544d6bb52e406729593a34de5b3548224a03641ba6b25b444ffe1f7818ae
-size 20812252
+oid sha256:963af37efbefc144f8778ddbbc5ebf1fe45f3edcf5fd12a33eb320f0b99b1815
+size 20815002

variables/variables.data-00000-of-00001

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c7ad5dd5dcb707924bb4c019aa742e058cb68ebc4d835b9e300f6aa030fc5f72
+oid sha256:afda21fef2f9644ba58522c05cc6c29fcf9bb93b8ec6a86debb8a3e895204a8c
 size 3438636228