-
Notifications
You must be signed in to change notification settings - Fork 2
/
final_config.ini
29 lines (28 loc) · 840 Bytes
/
final_config.ini
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
[agent]
imagefile_path = /data/padl19t1/600_1_signs_1_words/train_resized/image_locations.txt
boxfile_path = /data/padl19t1/600_1_signs_1_words/train_resized/bounding_boxes.npy
resultdir_path = /data/padl19t1/results_sparse_reward_complicated_data
agentdir_path = /data/padl19t1/agent_sparse_reward_complicated_data
gpu_id = 1
epsilon = 0.01
learning_rate = 0.0001
gamma = 0.95
replay_start_size = 100
replay_buffer_capacity = 20000
update_interval = 1
target_update_interval = 100
start_epsilon = 1.0
end_epsilon = 0.1
decay_steps = 300000
steps = 500000
train_max_episode_len = 100
eval_n_episodes = 10
eval_interval = 500
use_tensorboard = False
save_eval = False
pred_bboxes = ./pred_bboxes.npy
pred_labels = ./pred_labels.npy
pred_scores = ./pred_scores.npy
gt_bboxes = ./gt_bboxes.npy
gt_labels = ./gt_labels.npy
iou_threshold = 0.5