File size: 416 Bytes
6342c7b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
{
  "class": "C2CProjector",
  "init_args": {
    "source_dim": 128,
    "target_dim": 128,
    "source_num_heads": 8,
    "target_num_heads": 8,
    "hidden_dim": 1024,
    "intermediate_dim": 1024,
    "num_layers": 3,
    "dropout": 0.1,
    "initial_temperature": 1.0,
    "final_temperature": 0.001,
    "anneal_steps": 1953,
    "dtype": {
      "__type__": "torch.dtype",
      "value": "bfloat16"
    }
  }
}