File size: 650 Bytes
e3b38a0 | 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 45 46 47 48 | {
"model_class": "rl.models.GATV2ActorCritic",
"format": "raw PyTorch state_dict",
"n_node_features": 16,
"n_edge_features": 2,
"proj_out": 64,
"num_gat_blocks": 4,
"gat_channels": [
128,
128,
64,
64
],
"num_heads": [
8,
8,
4,
4
],
"concat": false,
"activation": "tanh",
"attn_dropout": [
0.0,
0.0,
0.0,
0.0
],
"feat_dropout": [
0.0,
0.0,
0.0,
0.0
],
"actor_head_dropout": 0.0,
"critic_head_dropout": 0.0,
"actor_head_layers": [
256,
128,
64
],
"critic_head_layers": [
256,
128,
64
],
"critic_readout_type": "sum"
}
|