Push Keras model using huggingface_hub.

.gitattributes

@@ -32,3 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+variables/variables.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,45 @@
+---
+library_name: keras
+---
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+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 | False |
+| 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

@@ -0,0 +1 @@
+{"name": "diffusion_model_4", "layers": [{"class_name": "InputLayer", "config": {"batch_input_shape": [null, 320], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_16"}, "name": "input_16", "inbound_nodes": []}, {"class_name": "Dense", "config": {"name": "dense_880", "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_880", "inbound_nodes": [[["input_16", 0, 0, {}]]]}, {"class_name": "InputLayer", "config": {"batch_input_shape": [null, 64, 64, 4], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_17"}, "name": "input_17", "inbound_nodes": []}, {"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": [[["dense_880", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_396", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "kernel_size": 3, "padding": 1, "filters": 320}, "name": "padded_conv2d_396", "inbound_nodes": [[["input_17", 0, 0, {}]]]}, {"class_name": "Dense", "config": {"name": "dense_881", "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_881", "inbound_nodes": [[["activation_274", 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_396", 0, 0, {}], ["dense_881", 0, 0, {}]]]}, {"class_name": "InputLayer", "config": {"batch_input_shape": [null, 77, 768], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_15"}, "name": "input_15", "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_15", 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_881", 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_15", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_405", "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_405", "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_405", 0, 0, {}], ["dense_881", 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_15", 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_881", 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_15", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_414", "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_414", "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_414", 0, 0, {}], ["dense_881", 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_15", 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_881", 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_15", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_423", "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_423", "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_423", 0, 0, {}], ["dense_881", 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_881", 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_881", 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_15", 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_881", 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_881", 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_881", 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_423", 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_881", 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_881", 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_15", 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_881", 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_15", 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_414", 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_881", 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_15", 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_881", 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_15", 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_881", 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_15", 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_405", 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_881", 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_15", 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_881", 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_15", 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_881", 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_15", 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_396", 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_881", 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_15", 0, 0, {}]]]}, {"class_name": "GroupNormalization", "config": {"name": "group_normalization_348", "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_348", "inbound_nodes": [[["spatial_transformer_79", 0, 0, {}]]]}, {"class_name": "Activation", "config": {"name": "activation_341", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "activation": "swish"}, "name": "activation_341", "inbound_nodes": [[["group_normalization_348", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_479", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "kernel_size": 3, "padding": 1, "filters": 4}, "name": "padded_conv2d_479", "inbound_nodes": [[["activation_341", 0, 0, {}]]]}], "input_layers": [["input_17", 0, 0], ["input_16", 0, 0], ["input_15", 0, 0]], "output_layers": [["padded_conv2d_479", 0, 0]]}

fingerprint.pb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7c4ce63cc89b9050500e6646ec8ab4b195dcca41d41d3332fcea30e98dfb8c43
+size 53

keras_metadata.pb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1566b3c08785151cb70eabee852ae5ed72d523fd41f9a7a90d441af251bd6ee5
+size 861444

saved_model.pb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:22332cd6ff4a8d0e61c33ff56fb1c6bab71875fd382b5b97d130eaec36d6288b
+size 20927208

variables/variables.data-00000-of-00001

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ae550f33cfa905eead471f39323ebd95a91951b7aebaa8cb92e7eacf4a708767
+size 3438636695