File size: 271 Bytes
df60eb9
e066e3b
 
 
 
df60eb9
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
{
  "auto_map":{
    "AutoConfig": "model_configuration.ModelConfig" ,
    "AutoModel": "modelling.MyModel",
  }, 
  "dropout": 0.2,
  "embed_size": 256,
  "latent_dim": 16,
  "max_len": 128,
  "num_classes": 5,
  "num_heads": 2,
  "num_layers": 2,
  "vocab_size": 8192
}