-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy pathexample.ini
More file actions
35 lines (35 loc) · 712 Bytes
/
example.ini
File metadata and controls
35 lines (35 loc) · 712 Bytes
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
[agent]
# data and paths
boxfile_path=../generated_data/bounding_boxes.npy
imagefile_path=../generated_data/image_locations.txt
resultdir_path=./results
agentdir_path=./agent
# hardware
gpu_id=-1
# optimizer
epsilon=0.01
learning_rate=0.0001
# agent
gamma=0.1
replay_start_size=100
replay_buffer_capacity=20000
update_interval=1
target_update_interval=100
# explorer
start_epsilon=1.0
end_epsilon=0.1
decay_steps=300000
# training
steps=1000000
train_max_episode_len=100
eval_n_episodes=10
eval_interval=500
use_tensorboard=True
# eval
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