/
gobigger_no_spatial_config.py
executable file
·68 lines (67 loc) · 1.83 KB
/
gobigger_no_spatial_config.py
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
from easydict import EasyDict
gobigger_config = dict(
exp_name='gobigger_baseline_v030',
env=dict(
collector_env_num=8,
evaluator_env_num=3,
n_evaluator_episode=3,
stop_value=1e10,
team_num=4,
player_num_per_team=3,
match_time=60*10,
map_height=1000,
map_width=1000,
spatial=False,
speed = False,
all_vision = False,
manager=dict(shared_memory=False, ),
),
policy=dict(
cuda=True,
on_policy=False,
priority=False,
priority_IS_weight=False,
model=dict(
scalar_shape=5,
food_shape=2,
food_relation_shape=150,
thorn_relation_shape=12,
clone_shape=17,
clone_relation_shape=12,
hidden_shape=128,
encode_shape=32,
action_type_shape=16,
),
learn=dict(
update_per_collect=8,
batch_size=128,
learning_rate=0.001,
target_theta=0.005,
discount_factor=0.99,
ignore_done=False,
learner=dict(
hook=dict(save_ckpt_after_iter=1000)),
),
collect=dict(n_sample=512, unroll_len=1, alpha=1.0),
eval=dict(evaluator=dict(eval_freq=1000,)),
other=dict(
eps=dict(
type='exp',
start=0.95,
end=0.5,
decay=100000,
),
replay_buffer=dict(replay_buffer_size=100000, ),
),
),
)
main_config = EasyDict(gobigger_config)
gobigger_create_config = dict(
env=dict(
type='gobigger',
import_names=['dizoo.gobigger.envs.gobigger_env'],
),
env_manager=dict(type='subprocess'),
policy=dict(type='dqn'),
)
create_config = EasyDict(gobigger_create_config)