-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathconfig.yml
49 lines (40 loc) · 1.41 KB
/
config.yml
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
###############################################################################
# SET GENERAL PARAMETERS
###############################################################################
ATARI_ENV: PongDeterministic-v4
RENDER: False
STEPS: 1000000
VERBOSE_STEPS: 1000
WARM_UP_STEPS: 1
SUMMARY_PATH: logs
###############################################################################
# SET GENERATOR PARAMETERS
###############################################################################
GEN_DATA_PATH: data
###############################################################################
# SET OFFLINE TRAINING PARAMETERS
###############################################################################
AGENT: REM
TRAIN_DATA_PATH: data/DQN_1m
EPOCHS: 200
BATCH_SIZE: 256
EVAL_EPISODES: 10
EVAL_RENDER: False
LEARNING_RATE: 0.0005
GAMMA: 0.99
NUM_HEADS: 64
TARGET_UPDATE_INTERVAL: 2500
SUMMARY_CHECKPOINT: 1000
###############################################################################
# SET INFERENCE PARAMETERS
###############################################################################
NUM_ENVS: 4
INF_AGENT: REM
INF_MODEL: models/REMOffDQNAgent/5.pt
###############################################################################
# SET AUTOENCODER PARAMETERS
###############################################################################
AUTO_GENERATE: True
AUTO_SAVE_PATH: autoencoder
AUTO_EPOCHS: 250
AUTO_BATCH_SIZE: 128