Push Keras model using huggingface_hub.

README.md

@@ -13,3 +13,33 @@ More information needed
 ## Training and evaluation data
 
 More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+
+| Hyperparameters | Value |
+| :-- | :-- |
+| inner_optimizer.class_name | Custom>RMSprop |
+| inner_optimizer.config.name | RMSprop |
+| inner_optimizer.config.weight_decay | None |
+| inner_optimizer.config.clipnorm | None |
+| inner_optimizer.config.global_clipnorm | None |
+| inner_optimizer.config.clipvalue | None |
+| inner_optimizer.config.use_ema | False |
+| inner_optimizer.config.ema_momentum | 0.99 |
+| inner_optimizer.config.ema_overwrite_frequency | 100 |
+| inner_optimizer.config.jit_compile | True |
+| inner_optimizer.config.is_legacy_optimizer | False |
+| inner_optimizer.config.learning_rate | 0.0010000000474974513 |
+| inner_optimizer.config.rho | 0.9 |
+| inner_optimizer.config.momentum | 0.0 |
+| inner_optimizer.config.epsilon | 1e-07 |
+| inner_optimizer.config.centered | False |
+| dynamic | True |
+| initial_scale | 32768.0 |
+| dynamic_growth_steps | 2000 |
+| training_precision | mixed_float16 |
+

config.json

@@ -1 +1 @@
-{"name": "diffusion_model", "layers": [{"class_name": "InputLayer", "config": {"batch_input_shape": [null, 320], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_3"}, "name": "input_3", "inbound_nodes": []}, {"class_name": "Dense", "config": {"name": "dense_72", "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_72", "inbound_nodes": [[["input_3", 0, 0, {}]]]}, {"class_name": "InputLayer", "config": {"batch_input_shape": [null, 64, 64, 4], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_4"}, "name": "input_4", "inbound_nodes": []}, {"class_name": "Activation", "config": {"name": "activation_1", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "activation": "swish"}, "name": "activation_1", "inbound_nodes": [[["dense_72", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_30", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "kernel_size": 3, "padding": 1, "filters": 320}, "name": "padded_conv2d_30", "inbound_nodes": [[["input_4", 0, 0, {}]]]}, {"class_name": "Dense", "config": {"name": "dense_73", "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_73", "inbound_nodes": [[["activation_1", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block", "inbound_nodes": [[["padded_conv2d_30", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "InputLayer", "config": {"batch_input_shape": [null, 77, 768], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_2"}, "name": "input_2", "inbound_nodes": []}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer", "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", "inbound_nodes": [[["res_block", 0, 0, {}], ["input_2", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_1", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_1", "inbound_nodes": [[["spatial_transformer", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_1", "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_1", "inbound_nodes": [[["res_block_1", 0, 0, {}], ["input_2", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_39", "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_39", "inbound_nodes": [[["spatial_transformer_1", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_2", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_2", "inbound_nodes": [[["padded_conv2d_39", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_2", "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_2", "inbound_nodes": [[["res_block_2", 0, 0, {}], ["input_2", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_3", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_3", "inbound_nodes": [[["spatial_transformer_2", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_3", "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_3", "inbound_nodes": [[["res_block_3", 0, 0, {}], ["input_2", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_48", "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_48", "inbound_nodes": [[["spatial_transformer_3", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_4", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_4", "inbound_nodes": [[["padded_conv2d_48", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_4", "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_4", "inbound_nodes": [[["res_block_4", 0, 0, {}], ["input_2", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_5", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_5", "inbound_nodes": [[["spatial_transformer_4", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_5", "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_5", "inbound_nodes": [[["res_block_5", 0, 0, {}], ["input_2", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_57", "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_57", "inbound_nodes": [[["spatial_transformer_5", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_6", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_6", "inbound_nodes": [[["padded_conv2d_57", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_7", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_7", "inbound_nodes": [[["res_block_6", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_8", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_8", "inbound_nodes": [[["res_block_7", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_6", "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_6", "inbound_nodes": [[["res_block_8", 0, 0, {}], ["input_2", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_9", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_9", "inbound_nodes": [[["spatial_transformer_6", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate", "inbound_nodes": [[["res_block_9", 0, 0, {}], ["res_block_7", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_10", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_10", "inbound_nodes": [[["concatenate", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_1", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_1", "inbound_nodes": [[["res_block_10", 0, 0, {}], ["res_block_6", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_11", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_11", "inbound_nodes": [[["concatenate_1", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_2", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_2", "inbound_nodes": [[["res_block_11", 0, 0, {}], ["padded_conv2d_57", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_12", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_12", "inbound_nodes": [[["concatenate_2", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "Upsample", "config": {"name": "upsample", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "channels": 1280}, "name": "upsample", "inbound_nodes": [[["res_block_12", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_3", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_3", "inbound_nodes": [[["upsample", 0, 0, {}], ["spatial_transformer_5", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_13", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_13", "inbound_nodes": [[["concatenate_3", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_7", "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_7", "inbound_nodes": [[["res_block_13", 0, 0, {}], ["input_2", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_4", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_4", "inbound_nodes": [[["spatial_transformer_7", 0, 0, {}], ["spatial_transformer_4", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_14", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_14", "inbound_nodes": [[["concatenate_4", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_8", "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_8", "inbound_nodes": [[["res_block_14", 0, 0, {}], ["input_2", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_5", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_5", "inbound_nodes": [[["spatial_transformer_8", 0, 0, {}], ["padded_conv2d_48", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_15", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_15", "inbound_nodes": [[["concatenate_5", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_9", "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_9", "inbound_nodes": [[["res_block_15", 0, 0, {}], ["input_2", 0, 0, {}]]]}, {"class_name": "Upsample", "config": {"name": "upsample_1", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "channels": 1280}, "name": "upsample_1", "inbound_nodes": [[["spatial_transformer_9", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_6", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_6", "inbound_nodes": [[["upsample_1", 0, 0, {}], ["spatial_transformer_3", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_16", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_16", "inbound_nodes": [[["concatenate_6", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_10", "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_10", "inbound_nodes": [[["res_block_16", 0, 0, {}], ["input_2", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_7", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_7", "inbound_nodes": [[["spatial_transformer_10", 0, 0, {}], ["spatial_transformer_2", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_17", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_17", "inbound_nodes": [[["concatenate_7", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_11", "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_11", "inbound_nodes": [[["res_block_17", 0, 0, {}], ["input_2", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_8", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_8", "inbound_nodes": [[["spatial_transformer_11", 0, 0, {}], ["padded_conv2d_39", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_18", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_18", "inbound_nodes": [[["concatenate_8", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_12", "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_12", "inbound_nodes": [[["res_block_18", 0, 0, {}], ["input_2", 0, 0, {}]]]}, {"class_name": "Upsample", "config": {"name": "upsample_2", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "channels": 640}, "name": "upsample_2", "inbound_nodes": [[["spatial_transformer_12", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_9", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_9", "inbound_nodes": [[["upsample_2", 0, 0, {}], ["spatial_transformer_1", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_19", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_19", "inbound_nodes": [[["concatenate_9", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_13", "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_13", "inbound_nodes": [[["res_block_19", 0, 0, {}], ["input_2", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_10", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_10", "inbound_nodes": [[["spatial_transformer_13", 0, 0, {}], ["spatial_transformer", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_20", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_20", "inbound_nodes": [[["concatenate_10", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_14", "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_14", "inbound_nodes": [[["res_block_20", 0, 0, {}], ["input_2", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_11", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_11", "inbound_nodes": [[["spatial_transformer_14", 0, 0, {}], ["padded_conv2d_30", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_21", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_21", "inbound_nodes": [[["concatenate_11", 0, 0, {}], ["dense_73", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_15", "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_15", "inbound_nodes": [[["res_block_21", 0, 0, {}], ["input_2", 0, 0, {}]]]}, {"class_name": "GroupNormalization", "config": {"name": "group_normalization_82", "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_82", "inbound_nodes": [[["spatial_transformer_15", 0, 0, {}]]]}, {"class_name": "Activation", "config": {"name": "activation_68", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "activation": "swish"}, "name": "activation_68", "inbound_nodes": [[["group_normalization_82", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_113", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "kernel_size": 3, "padding": 1, "filters": 4}, "name": "padded_conv2d_113", "inbound_nodes": [[["activation_68", 0, 0, {}]]]}], "input_layers": [["input_4", 0, 0], ["input_3", 0, 0], ["input_2", 0, 0]], "output_layers": [["padded_conv2d_113", 0, 0]]}
+{"name": "diffusion_model_1", "layers": [{"class_name": "InputLayer", "config": {"batch_input_shape": [null, 320], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_7"}, "name": "input_7", "inbound_nodes": []}, {"class_name": "Dense", "config": {"name": "dense_328", "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_328", "inbound_nodes": [[["input_7", 0, 0, {}]]]}, {"class_name": "InputLayer", "config": {"batch_input_shape": [null, 64, 64, 4], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_8"}, "name": "input_8", "inbound_nodes": []}, {"class_name": "Activation", "config": {"name": "activation_70", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "activation": "swish"}, "name": "activation_70", "inbound_nodes": [[["dense_328", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_152", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "kernel_size": 3, "padding": 1, "filters": 320}, "name": "padded_conv2d_152", "inbound_nodes": [[["input_8", 0, 0, {}]]]}, {"class_name": "Dense", "config": {"name": "dense_329", "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_329", "inbound_nodes": [[["activation_70", 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_152", 0, 0, {}], ["dense_329", 0, 0, {}]]]}, {"class_name": "InputLayer", "config": {"batch_input_shape": [null, 77, 768], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_6"}, "name": "input_6", "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_6", 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_329", 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_6", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_161", "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_161", "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_161", 0, 0, {}], ["dense_329", 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_6", 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_329", 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_6", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_170", "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_170", "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_170", 0, 0, {}], ["dense_329", 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_6", 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_329", 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_6", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_179", "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_179", "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_179", 0, 0, {}], ["dense_329", 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_329", 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_329", 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_6", 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_329", 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_329", 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_329", 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_179", 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_329", 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_329", 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_6", 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_329", 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_6", 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_170", 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_329", 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_6", 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_329", 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_6", 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_329", 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_6", 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_161", 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_329", 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_6", 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_329", 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_6", 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_329", 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_6", 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_152", 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_329", 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_6", 0, 0, {}]]]}, {"class_name": "GroupNormalization", "config": {"name": "group_normalization_173", "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_173", "inbound_nodes": [[["spatial_transformer_31", 0, 0, {}]]]}, {"class_name": "Activation", "config": {"name": "activation_137", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "activation": "swish"}, "name": "activation_137", "inbound_nodes": [[["group_normalization_173", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_235", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "kernel_size": 3, "padding": 1, "filters": 4}, "name": "padded_conv2d_235", "inbound_nodes": [[["activation_137", 0, 0, {}]]]}], "input_layers": [["input_8", 0, 0], ["input_7", 0, 0], ["input_6", 0, 0]], "output_layers": [["padded_conv2d_235", 0, 0]]}

fingerprint.pb

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:be83f6d66e2f66c01976eda6c12fb2fb20b36f4290a3556497f97f7e6f2e9caa
+oid sha256:a9bf3800141022fd81277b783fded09e7eadf8dd5bc634f55ffbe28accc7c162
 size 54

keras_metadata.pb

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4a08bbc66fe6db483bc2ee59160702c0eb6f2c2cbcb3e483ca5b51a254d6a4d4
-size 859407
+oid sha256:c448c699829b6703a19b7ea808dd4b2e82ac52cad2a5d6e44185cf377869c06f
+size 860974

saved_model.pb

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7e6247880367c464b2aee07a6fb177cb66d4ea005e5db438716429ebaf8e2b07
-size 20514784
+oid sha256:2ba01d8adf6fd88e7f60953068669ec8825f85b3e27cbe0cf80ad7e094fed5a8
+size 20835763

variables/variables.data-00000-of-00001

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c81a5bd1b260684068fd4b8ac25a76873fabed29e1d6318c3645b846c48ef97c
-size 3438634699
+oid sha256:e5a609687e99cd410c6d0a29b6dbd1cd31801f300d7836037bb87b5413d31e8b
+size 3438636312