beat-encoder / config.json
bdhanani's picture
Upload config.json with huggingface_hub
f319e60 verified
Raw
History Blame Contribute Delete
633 Bytes
{
"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"
]
}