File size: 1,060 Bytes
ede8cce
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
{
    "obs_dim": 358,
    "action_dim": 69,
    "device": "cpu",
    "archi": {
        "z_dim": 256,
        "norm_z": true,
        "f": {
            "hidden_dim": 1024,
            "model": "simple",
            "hidden_layers": 2,
            "embedding_layers": 2,
            "num_parallel": 2,
            "ensemble_mode": "batch"
        },
        "b": {
            "hidden_dim": 256,
            "hidden_layers": 1,
            "norm": true
        },
        "actor": {
            "hidden_dim": 1024,
            "model": "simple",
            "hidden_layers": 2,
            "embedding_layers": 2
        },
        "critic": {
            "hidden_dim": 1024,
            "model": "simple",
            "hidden_layers": 2,
            "embedding_layers": 2,
            "num_parallel": 2,
            "ensemble_mode": "batch"
        },
        "discriminator": {
            "hidden_dim": 1024,
            "hidden_layers": 3
        }
    },
    "inference_batch_size": 500000,
    "seq_length": 8,
    "actor_std": 0.2,
    "norm_obs": true
}