-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfigs.yaml
32 lines (31 loc) · 1.11 KB
/
configs.yaml
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
# train
# python train.py --algos=dtqn_noisy --self_play_episode_num=4 --load_ckpt=True --save_ckpt=True --data_folder="train_data_/benchmark_reduced" --wandbName="NoPer" --hid_layer=3 --emb_dim=64 --context_len=5 --early_stop=False --enable_wandb=True
# eval
# python train.py --algos=dtqn_noisy --self_play_episode_num=150 --result_dir=solutionsDRL --load_ckpt=True --save_ckpt=False --data_folder="test_data_/benchmark_reduced" --wandbName="dtqn_noisy_pretain_test" --hid_layer=3 --emb_dim=64 --context_len=5
default:
all:
algos: None
wandbName: None
hid_layer: &lay 3
emb_dim: &dim 64 # 128 is a little too large...
context_len: &ctx_l 5
early_stop: False
result_dir: solutionsDRL
train:
mode: train
load_ckpt: True
save_ckpt: True
self_play_episode_num: 4
run_benchmark_num: 30
enable_wandb: True
data_folder: train_data_/benchmark_reduced
eval:
mode: eval
load_ckpt: True
save_ckpt: False
self_play_episode_num: 150
enable_wandb: True
data_folder: test_data_/benchmark_reduced
# custom:
# dtqn:
# algos