| { | |
| "alphabet_size": 7, | |
| "attention_heads": 16, | |
| "attention_maps_to_save": [], | |
| "embed_dim": 1024, | |
| "embeddings_layers_to_save": [], | |
| "ffn_embed_dim": 2048, | |
| "interpolation_method": "nearest", | |
| "key_size": 64, | |
| "layer_norm_eps": 1e-05, | |
| "mask_token_id": 5, | |
| "max_positions": 20480, | |
| "num_cells": 10, | |
| "num_downsamples": 8, | |
| "num_hidden_layers_head": 1, | |
| "num_layers": 4, | |
| "num_scales": 10, | |
| "pad_token_id": 6, | |
| "sigma_max": 10.0, | |
| "sigma_min": 1.0, | |
| "token_embed_dim": 64, | |
| "use_gradient_checkpointing": false, | |
| "use_skip_connection": true, | |
| "use_spatial_information": false | |
| } |