forked from johnjim0816/joyrl-offline
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.py
31 lines (30 loc) · 1.06 KB
/
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
#!/usr/bin/env python
# coding=utf-8
'''
@Author: John
@Email: [email protected]
@Date: 2020-06-12 00:50:49
@LastEditor: John
LastEditTime: 2023-05-18 13:14:15
@Discription:
@Environment: python 3.7.7
'''
from config.config import DefaultConfig
class AlgoConfig(DefaultConfig):
''' algorithm parameters
'''
def __init__(self) -> None:
# set epsilon_start=epsilon_end can obtain fixed epsilon=epsilon_end
self.epsilon_start = 0.95 # epsilon start value
self.epsilon_end = 0.01 # epsilon end value
self.epsilon_decay = 500 # epsilon decay rate
self.gamma = 0.99 # discount factor
self.lr = 0.0001 # learning rate
self.buffer_size = 100000 # size of replay buffer
self.batch_size = 64 # batch size
self.target_update = 4 # target network update frequency
# value network layers config
self.value_layers = [
{'layer_type': 'Linear', 'layer_size': [256], 'activation': 'ReLU'},
{'layer_type': 'Linear', 'layer_size': [256], 'activation': 'ReLU'},
]