Push Keras model using huggingface_hub.

README.md

@@ -1,24 +1,14 @@
 ---
 library_name: keras
-license: creativeml-openrail-m
-tags:
-- stable-diffusion
-- text-to-image
-- keras-dreambooth
-- sci-fi
-- fantasy
-pipeline_tag: text-to-image
 ---
 
 ## Model description
 
-This is a Keras Dreambooth model fine-tuned to Hogwarts Legacy game characters with custom gears in their outfit, different from base SD model if the prompt is "hogwarts student".
-
-<img src="https://s3.amazonaws.com/moonup/production/uploads/6329449376bcaa47f64a2e63/dVjwmPH1HzpIIs6j_vnOR.png" width="538" height="318"/>
+More information needed
 
 ## Intended uses & limitations
 
-The base prompt is "hogwarts [legacy] student". A slightly better prompt would be "a digital art of a hogwarts [legacy] student in a forest, high quality" with negative prompt "distorted".
+More information needed
 
 ## Training and evaluation data
 
@@ -51,4 +41,5 @@ The following hyperparameters were used during training:
 | dynamic | True |
 | initial_scale | 32768.0 |
 | dynamic_growth_steps | 2000 |
-| training_precision | mixed_float16 |
+| training_precision | mixed_float16 |
+

config.json

@@ -1 +1 @@
-{"name": "diffusion_model_3", "layers": [{"class_name": "InputLayer", "config": {"batch_input_shape": [null, 320], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_14"}, "name": "input_14", "inbound_nodes": []}, {"class_name": "Dense", "config": {"name": "dense_696", "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_696", "inbound_nodes": [[["input_14", 0, 0, {}]]]}, {"class_name": "InputLayer", "config": {"batch_input_shape": [null, 64, 64, 4], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_15"}, "name": "input_15", "inbound_nodes": []}, {"class_name": "Activation", "config": {"name": "activation_207", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "activation": "swish"}, "name": "activation_207", "inbound_nodes": [[["dense_696", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_350", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "kernel_size": 3, "padding": 1, "filters": 320}, "name": "padded_conv2d_350", "inbound_nodes": [[["input_15", 0, 0, {}]]]}, {"class_name": "Dense", "config": {"name": "dense_697", "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_697", "inbound_nodes": [[["activation_207", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_66", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_66", "inbound_nodes": [[["padded_conv2d_350", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "InputLayer", "config": {"batch_input_shape": [null, 77, 768], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_13"}, "name": "input_13", "inbound_nodes": []}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_48", "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_48", "inbound_nodes": [[["res_block_66", 0, 0, {}], ["input_13", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_67", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_67", "inbound_nodes": [[["spatial_transformer_48", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_49", "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_49", "inbound_nodes": [[["res_block_67", 0, 0, {}], ["input_13", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_359", "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_359", "inbound_nodes": [[["spatial_transformer_49", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_68", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_68", "inbound_nodes": [[["padded_conv2d_359", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_50", "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_50", "inbound_nodes": [[["res_block_68", 0, 0, {}], ["input_13", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_69", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_69", "inbound_nodes": [[["spatial_transformer_50", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_51", "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_51", "inbound_nodes": [[["res_block_69", 0, 0, {}], ["input_13", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_368", "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_368", "inbound_nodes": [[["spatial_transformer_51", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_70", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_70", "inbound_nodes": [[["padded_conv2d_368", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_52", "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_52", "inbound_nodes": [[["res_block_70", 0, 0, {}], ["input_13", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_71", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_71", "inbound_nodes": [[["spatial_transformer_52", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_53", "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_53", "inbound_nodes": [[["res_block_71", 0, 0, {}], ["input_13", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_377", "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_377", "inbound_nodes": [[["spatial_transformer_53", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_72", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_72", "inbound_nodes": [[["padded_conv2d_377", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_73", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_73", "inbound_nodes": [[["res_block_72", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_74", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_74", "inbound_nodes": [[["res_block_73", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_54", "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_54", "inbound_nodes": [[["res_block_74", 0, 0, {}], ["input_13", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_75", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_75", "inbound_nodes": [[["spatial_transformer_54", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_36", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_36", "inbound_nodes": [[["res_block_75", 0, 0, {}], ["res_block_73", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_76", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_76", "inbound_nodes": [[["concatenate_36", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_37", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_37", "inbound_nodes": [[["res_block_76", 0, 0, {}], ["res_block_72", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_77", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_77", "inbound_nodes": [[["concatenate_37", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_38", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_38", "inbound_nodes": [[["res_block_77", 0, 0, {}], ["padded_conv2d_377", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_78", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_78", "inbound_nodes": [[["concatenate_38", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "Upsample", "config": {"name": "upsample_9", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "channels": 1280}, "name": "upsample_9", "inbound_nodes": [[["res_block_78", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_39", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_39", "inbound_nodes": [[["upsample_9", 0, 0, {}], ["spatial_transformer_53", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_79", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_79", "inbound_nodes": [[["concatenate_39", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_55", "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_55", "inbound_nodes": [[["res_block_79", 0, 0, {}], ["input_13", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_40", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_40", "inbound_nodes": [[["spatial_transformer_55", 0, 0, {}], ["spatial_transformer_52", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_80", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_80", "inbound_nodes": [[["concatenate_40", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_56", "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_56", "inbound_nodes": [[["res_block_80", 0, 0, {}], ["input_13", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_41", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_41", "inbound_nodes": [[["spatial_transformer_56", 0, 0, {}], ["padded_conv2d_368", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_81", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_81", "inbound_nodes": [[["concatenate_41", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_57", "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_57", "inbound_nodes": [[["res_block_81", 0, 0, {}], ["input_13", 0, 0, {}]]]}, {"class_name": "Upsample", "config": {"name": "upsample_10", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "channels": 1280}, "name": "upsample_10", "inbound_nodes": [[["spatial_transformer_57", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_42", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_42", "inbound_nodes": [[["upsample_10", 0, 0, {}], ["spatial_transformer_51", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_82", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_82", "inbound_nodes": [[["concatenate_42", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_58", "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_58", "inbound_nodes": [[["res_block_82", 0, 0, {}], ["input_13", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_43", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_43", "inbound_nodes": [[["spatial_transformer_58", 0, 0, {}], ["spatial_transformer_50", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_83", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_83", "inbound_nodes": [[["concatenate_43", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_59", "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_59", "inbound_nodes": [[["res_block_83", 0, 0, {}], ["input_13", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_44", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_44", "inbound_nodes": [[["spatial_transformer_59", 0, 0, {}], ["padded_conv2d_359", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_84", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_84", "inbound_nodes": [[["concatenate_44", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_60", "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_60", "inbound_nodes": [[["res_block_84", 0, 0, {}], ["input_13", 0, 0, {}]]]}, {"class_name": "Upsample", "config": {"name": "upsample_11", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "channels": 640}, "name": "upsample_11", "inbound_nodes": [[["spatial_transformer_60", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_45", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_45", "inbound_nodes": [[["upsample_11", 0, 0, {}], ["spatial_transformer_49", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_85", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_85", "inbound_nodes": [[["concatenate_45", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_61", "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_61", "inbound_nodes": [[["res_block_85", 0, 0, {}], ["input_13", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_46", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_46", "inbound_nodes": [[["spatial_transformer_61", 0, 0, {}], ["spatial_transformer_48", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_86", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_86", "inbound_nodes": [[["concatenate_46", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_62", "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_62", "inbound_nodes": [[["res_block_86", 0, 0, {}], ["input_13", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_47", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_47", "inbound_nodes": [[["spatial_transformer_62", 0, 0, {}], ["padded_conv2d_350", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_87", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_87", "inbound_nodes": [[["concatenate_47", 0, 0, {}], ["dense_697", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_63", "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_63", "inbound_nodes": [[["res_block_87", 0, 0, {}], ["input_13", 0, 0, {}]]]}, {"class_name": "GroupNormalization", "config": {"name": "group_normalization_317", "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_317", "inbound_nodes": [[["spatial_transformer_63", 0, 0, {}]]]}, {"class_name": "Activation", "config": {"name": "activation_274", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "activation": "swish"}, "name": "activation_274", "inbound_nodes": [[["group_normalization_317", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_433", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "kernel_size": 3, "padding": 1, "filters": 4}, "name": "padded_conv2d_433", "inbound_nodes": [[["activation_274", 0, 0, {}]]]}], "input_layers": [["input_15", 0, 0], ["input_14", 0, 0], ["input_13", 0, 0]], "output_layers": [["padded_conv2d_433", 0, 0]]}
+{"name": "diffusion_model_4", "layers": [{"class_name": "InputLayer", "config": {"batch_input_shape": [null, 320], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_18"}, "name": "input_18", "inbound_nodes": []}, {"class_name": "Dense", "config": {"name": "dense_1024", "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_1024", "inbound_nodes": [[["input_18", 0, 0, {}]]]}, {"class_name": "InputLayer", "config": {"batch_input_shape": [null, 64, 64, 4], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_19"}, "name": "input_19", "inbound_nodes": []}, {"class_name": "Activation", "config": {"name": "activation_276", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "activation": "swish"}, "name": "activation_276", "inbound_nodes": [[["dense_1024", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_472", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "kernel_size": 3, "padding": 1, "filters": 320}, "name": "padded_conv2d_472", "inbound_nodes": [[["input_19", 0, 0, {}]]]}, {"class_name": "Dense", "config": {"name": "dense_1025", "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_1025", "inbound_nodes": [[["activation_276", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_88", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_88", "inbound_nodes": [[["padded_conv2d_472", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "InputLayer", "config": {"batch_input_shape": [null, 77, 768], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_17"}, "name": "input_17", "inbound_nodes": []}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_64", "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_64", "inbound_nodes": [[["res_block_88", 0, 0, {}], ["input_17", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_89", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_89", "inbound_nodes": [[["spatial_transformer_64", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_65", "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_65", "inbound_nodes": [[["res_block_89", 0, 0, {}], ["input_17", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_481", "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_481", "inbound_nodes": [[["spatial_transformer_65", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_90", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_90", "inbound_nodes": [[["padded_conv2d_481", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_66", "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_66", "inbound_nodes": [[["res_block_90", 0, 0, {}], ["input_17", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_91", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_91", "inbound_nodes": [[["spatial_transformer_66", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_67", "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_67", "inbound_nodes": [[["res_block_91", 0, 0, {}], ["input_17", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_490", "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_490", "inbound_nodes": [[["spatial_transformer_67", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_92", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_92", "inbound_nodes": [[["padded_conv2d_490", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_68", "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_68", "inbound_nodes": [[["res_block_92", 0, 0, {}], ["input_17", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_93", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_93", "inbound_nodes": [[["spatial_transformer_68", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_69", "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_69", "inbound_nodes": [[["res_block_93", 0, 0, {}], ["input_17", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_499", "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_499", "inbound_nodes": [[["spatial_transformer_69", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_94", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_94", "inbound_nodes": [[["padded_conv2d_499", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_95", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_95", "inbound_nodes": [[["res_block_94", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_96", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_96", "inbound_nodes": [[["res_block_95", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_70", "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_70", "inbound_nodes": [[["res_block_96", 0, 0, {}], ["input_17", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_97", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_97", "inbound_nodes": [[["spatial_transformer_70", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_48", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_48", "inbound_nodes": [[["res_block_97", 0, 0, {}], ["res_block_95", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_98", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_98", "inbound_nodes": [[["concatenate_48", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_49", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_49", "inbound_nodes": [[["res_block_98", 0, 0, {}], ["res_block_94", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_99", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_99", "inbound_nodes": [[["concatenate_49", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_50", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_50", "inbound_nodes": [[["res_block_99", 0, 0, {}], ["padded_conv2d_499", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_100", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_100", "inbound_nodes": [[["concatenate_50", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "Upsample", "config": {"name": "upsample_12", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "channels": 1280}, "name": "upsample_12", "inbound_nodes": [[["res_block_100", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_51", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_51", "inbound_nodes": [[["upsample_12", 0, 0, {}], ["spatial_transformer_69", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_101", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_101", "inbound_nodes": [[["concatenate_51", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_71", "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_71", "inbound_nodes": [[["res_block_101", 0, 0, {}], ["input_17", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_52", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_52", "inbound_nodes": [[["spatial_transformer_71", 0, 0, {}], ["spatial_transformer_68", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_102", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_102", "inbound_nodes": [[["concatenate_52", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_72", "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_72", "inbound_nodes": [[["res_block_102", 0, 0, {}], ["input_17", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_53", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_53", "inbound_nodes": [[["spatial_transformer_72", 0, 0, {}], ["padded_conv2d_490", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_103", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 1280}, "name": "res_block_103", "inbound_nodes": [[["concatenate_53", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_73", "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_73", "inbound_nodes": [[["res_block_103", 0, 0, {}], ["input_17", 0, 0, {}]]]}, {"class_name": "Upsample", "config": {"name": "upsample_13", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "channels": 1280}, "name": "upsample_13", "inbound_nodes": [[["spatial_transformer_73", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_54", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_54", "inbound_nodes": [[["upsample_13", 0, 0, {}], ["spatial_transformer_67", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_104", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_104", "inbound_nodes": [[["concatenate_54", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_74", "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_74", "inbound_nodes": [[["res_block_104", 0, 0, {}], ["input_17", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_55", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_55", "inbound_nodes": [[["spatial_transformer_74", 0, 0, {}], ["spatial_transformer_66", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_105", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_105", "inbound_nodes": [[["concatenate_55", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_75", "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_75", "inbound_nodes": [[["res_block_105", 0, 0, {}], ["input_17", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_56", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_56", "inbound_nodes": [[["spatial_transformer_75", 0, 0, {}], ["padded_conv2d_481", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_106", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 640}, "name": "res_block_106", "inbound_nodes": [[["concatenate_56", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_76", "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_76", "inbound_nodes": [[["res_block_106", 0, 0, {}], ["input_17", 0, 0, {}]]]}, {"class_name": "Upsample", "config": {"name": "upsample_14", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "channels": 640}, "name": "upsample_14", "inbound_nodes": [[["spatial_transformer_76", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_57", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_57", "inbound_nodes": [[["upsample_14", 0, 0, {}], ["spatial_transformer_65", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_107", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_107", "inbound_nodes": [[["concatenate_57", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_77", "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_77", "inbound_nodes": [[["res_block_107", 0, 0, {}], ["input_17", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_58", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_58", "inbound_nodes": [[["spatial_transformer_77", 0, 0, {}], ["spatial_transformer_64", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_108", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_108", "inbound_nodes": [[["concatenate_58", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_78", "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_78", "inbound_nodes": [[["res_block_108", 0, 0, {}], ["input_17", 0, 0, {}]]]}, {"class_name": "Concatenate", "config": {"name": "concatenate_59", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "axis": -1}, "name": "concatenate_59", "inbound_nodes": [[["spatial_transformer_78", 0, 0, {}], ["padded_conv2d_472", 0, 0, {}]]]}, {"class_name": "ResBlock", "config": {"name": "res_block_109", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "output_dim": 320}, "name": "res_block_109", "inbound_nodes": [[["concatenate_59", 0, 0, {}], ["dense_1025", 0, 0, {}]]]}, {"class_name": "SpatialTransformer", "config": {"name": "spatial_transformer_79", "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_79", "inbound_nodes": [[["res_block_109", 0, 0, {}], ["input_17", 0, 0, {}]]]}, {"class_name": "GroupNormalization", "config": {"name": "group_normalization_408", "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_408", "inbound_nodes": [[["spatial_transformer_79", 0, 0, {}]]]}, {"class_name": "Activation", "config": {"name": "activation_343", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "activation": "swish"}, "name": "activation_343", "inbound_nodes": [[["group_normalization_408", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_555", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "kernel_size": 3, "padding": 1, "filters": 4}, "name": "padded_conv2d_555", "inbound_nodes": [[["activation_343", 0, 0, {}]]]}], "input_layers": [["input_19", 0, 0], ["input_18", 0, 0], ["input_17", 0, 0]], "output_layers": [["padded_conv2d_555", 0, 0]]}

fingerprint.pb

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5d88f93c4e1be45ffb2136962dd22f98a3a407c840e1e4925cc6bfa678caac2d
+oid sha256:b0e949a775a30d13aa1e5a5383b888b531271a8e7021c0c3ef08bfdb82e1aa98
 size 55

keras_metadata.pb

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:63ea09a9c94eaef0670c96af50813c28c262fb76efaf267cca83f25660de5726
-size 861209
+oid sha256:52e7f86d8d4849b0eb09480be0e19f29523f4d465f41ac509dfa17be1c7c1789
+size 861761

saved_model.pb

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c87b913566a606382fb25e6bffcb2979181314cf78bba18f069607a411fc4013
-size 20895146
+oid sha256:97a36c050f27f84ae2f27fe11e5373d4800b9d574ed1013e5709e9005fc579db
+size 20964752

variables/variables.data-00000-of-00001

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c92e13b871b5849af7dfb84d6faf61db9f086b35c82c9d1afedfcbf218383a46
-size 3438636460
+oid sha256:6619ed71be51015eee5c558149f3ffc85d83e75b4189c82d009c83c2b9a95863
+size 3438636874