| { |
| "module": "keras_hub.src.models.parseq.parseq_causal_lm", |
| "class_name": "PARSeqCausalLM", |
| "config": { |
| "backbone": { |
| "module": "keras_hub.src.models.parseq.parseq_backbone", |
| "class_name": "PARSeqBackbone", |
| "config": { |
| "name": "par_seq_backbone", |
| "trainable": true, |
| "dtype": { |
| "module": "keras", |
| "class_name": "DTypePolicy", |
| "config": { |
| "name": "float32" |
| }, |
| "registered_name": null |
| }, |
| "image_encoder": { |
| "module": "keras_hub.src.models.vit.vit_backbone", |
| "class_name": "ViTBackbone", |
| "config": { |
| "name": "encoder", |
| "trainable": true, |
| "dtype": { |
| "module": "keras", |
| "class_name": "DTypePolicy", |
| "config": { |
| "name": "float32" |
| }, |
| "registered_name": null |
| }, |
| "image_shape": [ |
| 32, |
| 128, |
| 3 |
| ], |
| "patch_size": [ |
| 4, |
| 8 |
| ], |
| "num_layers": 12, |
| "num_heads": 6, |
| "hidden_dim": 384, |
| "mlp_dim": 1536, |
| "dropout_rate": 0.0, |
| "attention_dropout": 0.0, |
| "layer_norm_epsilon": 1e-06, |
| "use_mha_bias": true, |
| "use_mlp_bias": true, |
| "use_class_token": false, |
| "use_patch_bias": true |
| }, |
| "registered_name": "keras_hub>ViTBackbone" |
| }, |
| "vocabulary_size": 97, |
| "max_label_length": 25, |
| "decoder_hidden_dim": 384, |
| "num_decoder_layers": 1, |
| "num_decoder_heads": 12, |
| "decoder_mlp_dim": 1536, |
| "dropout_rate": 0.1, |
| "attention_dropout": 0.1 |
| }, |
| "registered_name": "keras_hub>PARSeqBackbone" |
| }, |
| "preprocessor": { |
| "module": "keras_hub.src.models.parseq.parseq_causal_lm_preprocessor", |
| "class_name": "PARSeqCausalLMPreprocessor", |
| "config": { |
| "name": "par_seq_causal_lm_preprocessor", |
| "trainable": true, |
| "dtype": { |
| "module": "keras", |
| "class_name": "DTypePolicy", |
| "config": { |
| "name": "float32" |
| }, |
| "registered_name": null |
| }, |
| "tokenizer": { |
| "module": "keras_hub.src.models.parseq.parseq_tokenizer", |
| "class_name": "PARSeqTokenizer", |
| "config": { |
| "name": "par_seq_tokenizer", |
| "trainable": true, |
| "dtype": { |
| "module": "keras", |
| "class_name": "DTypePolicy", |
| "config": { |
| "name": "int32" |
| }, |
| "registered_name": null |
| }, |
| "config_file": "tokenizer.json" |
| }, |
| "registered_name": "keras_hub>PARSeqTokenizer" |
| }, |
| "image_converter": { |
| "module": "keras_hub.src.models.parseq.parseq_image_converter", |
| "class_name": "PARSeqImageConverter", |
| "config": { |
| "name": "par_seq_image_converter", |
| "trainable": true, |
| "dtype": { |
| "module": "keras", |
| "class_name": "DTypePolicy", |
| "config": { |
| "name": "float32" |
| }, |
| "registered_name": null |
| }, |
| "image_size": [ |
| 32, |
| 128 |
| ], |
| "scale": 0.00784313725490196, |
| "offset": -1.0, |
| "interpolation": "bicubic", |
| "antialias": false, |
| "crop_to_aspect_ratio": true, |
| "pad_to_aspect_ratio": false, |
| "bounding_box_format": "yxyx" |
| }, |
| "registered_name": "keras_hub>PARSeqImageConverter" |
| }, |
| "config_file": "preprocessor.json", |
| "sequence_length": 25, |
| "add_start_token": true, |
| "add_end_token": true |
| }, |
| "registered_name": "keras_hub>PARSeqCausalLMPreprocessor" |
| }, |
| "name": "par_seq_causal_lm", |
| "num_perms": 6, |
| "add_forward_perms": true, |
| "add_mirrored_perms": true, |
| "seed": null, |
| "end_token_id": 0 |
| }, |
| "registered_name": "keras_hub>PARSeqCausalLM" |
| } |
