File size: 1,015 Bytes
079c32c |
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 |
from easydict import EasyDict
conv1d_config = dict(
feature_embedding=dict(
player=dict(
input_dim=36,
output_dim=64,
),
ball=dict(
input_dim=18,
output_dim=64,
),
left_team=dict(
input_dim=7,
output_dim=48,
conv1d_output_channel=36,
fc_output_dim=96,
),
right_team=dict(
input_dim=7,
output_dim=48,
conv1d_output_channel=36,
fc_output_dim=96,
),
left_closest=dict(
input_dim=7,
output_dim=48,
),
right_closest=dict(
input_dim=7,
output_dim=48,
)
),
fc_cat=dict(input_dim=416, ),
lstm_size=256,
policy_head=dict(
input_dim=256,
hidden_dim=164,
act_shape=19,
),
value_head=dict(input_dim=256, hidden_dim=164, output_dim=1),
)
conv1d_default_config = EasyDict(conv1d_config)
|