-
Notifications
You must be signed in to change notification settings - Fork 0
/
hparams.py
37 lines (29 loc) · 913 Bytes
/
hparams.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
import numpy as np
########### INPUT PARAMS ###########
FREQUENCY = 10 # loop frequency of the controller
FRAMERATE = 20 # framerate of the camera
HISTORY_LEN = 1
IMG_HEIGHT = 160
IMG_WIDTH = 240
IMG_DEPTH = 3 * HISTORY_LEN # 3 color channels, 4 frame history
FLOW_HEIGHT = 10
FLOW_WIDTH = 16
FLOW_DEPTH = 3 * HISTORY_LEN # 1 color channel, 4 frame history
MOTOR_DIM = 2 * HISTORY_LEN
SENSOR_DIM = 4 * HISTORY_LEN
########### OTHER STUFF ###########
DQN_URL = "http://agni:5000"
MOTOR_VALS = np.array([-0.2,0,0.2]) # forward, nothing, backward for each motor
ACTION_SPACE_SIZE = MOTOR_VALS.shape[0]**2
DISCOUNT_FACTOR = 0.95
#LEARNING_RATE = 1e-5 # TODO: May want to decay by using a tf.Variable
LEARNING_RATE = 0
BATCH_SIZE = 50
MEMORY_SIZE = 100
TARGET_Q_UPDATE_INTERVAL = 50
EPS=0.2
ACTION_TRAIN_LR = 3e-4
########### NET PARAMS ###########
IMG_CONV_LAYERS = 3
FLOW_CONV_LAYERS = 2
MOTOR_FC_LAYERS = 1