ENV_NAME = 'Unrealcv-Search-v0'

CONTINUE = False #load a pre-trained model
RESTART_EP = 3600 # the episode number of the pre-trained model

TRAIN = True # train the network
USE_TARGET_NETWORK = True # use the target network
SHOW = False # show the current state, reward and action
MAP = False # show the trajectory in 2d map

DEVICE_TF = '/gpu:0'
MAX_EPOCHS = 10000 # max episode number
MEMORY_SIZE = 20000
LEARN_START_STEP = 5000
INPUT_SIZE = 150
INPUT_CHANNELS = 3
BATCH_SIZE = 32
LEARNING_RATE = 1e-3  # 1e6
GAMMA = 0.95
INITIAL_EPSILON = 1  # starting value of epsilon
FINAL_EPSILON = 0.1  # final value of epsilon
MAX_EXPLORE_STEPS = 20000
TEST_INTERVAL_EPOCHS = 100
SAVE_INTERVAL_EPOCHS = 200

MONITOR_DIR = 'log/monitor/' #the path to save monitor file
MODEL_DIR = 'log/model' # the path to save deep model
PARAM_DIR = 'log/param' # the path to save the parameters
TRA_DIR = 'log/trajectory.csv' # the path to save trajectory

#the path to reload weights, monitor and params
weights_path = 'log/model/dqn_ep' + str(RESTART_EP)+ '.h5'
monitor_path = 'log/monitor/'+ str(RESTART_EP)
params_json = 'log/param/dqn_ep' + str(RESTART_EP) + '.json'
