| { | |
| "vocab_size": 512, | |
| "max_properties": 128, | |
| "max_seq_length": 256, | |
| "hidden_size": 768, | |
| "num_hidden_layers": 12, | |
| "num_attention_heads": 12, | |
| "intermediate_size": 3072, | |
| "hidden_dropout_prob": 0.1, | |
| "attention_probs_dropout_prob": 0.1, | |
| "time_embedding_size": 64, | |
| "max_time_gap": 90, | |
| "pooling_strategy": "cls", | |
| "masked_event_ratio": 0.15, | |
| "next_event_prediction": true, | |
| "contrastive_learning": true, | |
| "learning_rate": 0.0001, | |
| "weight_decay": 0.01, | |
| "warmup_steps": 1000, | |
| "max_steps": 100000, | |
| "batch_size": 16, | |
| "embedding_size": 768, | |
| "model_type": "beat", | |
| "architectures": [ | |
| "BEATModel" | |
| ] | |
| } |