| {"name": "model_3", "trainable": true, "layers": [{"module": "keras.layers", "class_name": "InputLayer", "config": {"batch_input_shape": [null, 32768], "dtype": "int32", "sparse": false, "ragged": false, "name": "input_4"}, "registered_name": null, "name": "input_4", "inbound_nodes": []}, {"module": "tokun.model", "class_name": "Encoder", "config": {"token_dim": [4, 16], "encoding_dim": 256, "embedding_dim": 256, "sequence_axis": 1, "feature_axis": -1, "activation": "gelu"}, "registered_name": "models>Encoder", "build_config": {"input_shape": [16, 2048]}, "name": "encoder_3", "inbound_nodes": [[["input_4", 0, 0, {}]]]}, {"module": "tokun.model", "class_name": "Decoder", "config": {"token_dim": [16, 4], "encoding_dim": 8, "embedding_dim": 256, "sequence_axis": 1, "feature_axis": -1, "activation": "gelu", "output": "binary"}, "registered_name": "models>Decoder", "build_config": {"input_shape": [16, 32, 256]}, "name": "decoder_3", "inbound_nodes": [[["encoder_3", 0, 0, {}]]]}], "input_layers": [["input_4", 0, 0]], "output_layers": [["decoder_3", 0, 0]]} |