| {"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_768", "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_768", "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_768", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_358", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "kernel_size": 3, "padding": 1, "filters": 320}, "name": "padded_conv2d_358", "inbound_nodes": [[["input_15", 0, 0, {}]]]}, {"class_name": "Dense", "config": {"name": "dense_769", "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_769", "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_358", 0, 0, {}], ["dense_769", 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_769", 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_367", "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_367", "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_367", 0, 0, {}], ["dense_769", 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_769", 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_376", "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_376", "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_376", 0, 0, {}], ["dense_769", 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_769", 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_385", "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_385", "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_385", 0, 0, {}], ["dense_769", 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_769", 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_769", 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_769", 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_769", 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_769", 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_385", 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_769", 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_769", 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_769", 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_376", 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_769", 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_769", 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_769", 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_367", 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_769", 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_769", 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_769", 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_358", 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_769", 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_325", "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_325", "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_325", 0, 0, {}]]]}, {"class_name": "PaddedConv2D", "config": {"name": "padded_conv2d_441", "trainable": true, "dtype": {"class_name": "Policy", "config": {"name": "mixed_float16"}, "shared_object_id": 1}, "kernel_size": 3, "padding": 1, "filters": 4}, "name": "padded_conv2d_441", "inbound_nodes": [[["activation_274", 0, 0, {}]]]}], "input_layers": [["input_15", 0, 0], ["input_14", 0, 0], ["input_13", 0, 0]], "output_layers": [["padded_conv2d_441", 0, 0]]} |