forked from mysbupt/FashionKE
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.yaml
28 lines (27 loc) · 870 Bytes
/
config.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
num_epoches: 20
batch_size: 20
batch_size_clean: 10
batch_size_noise: 10
test_batch_size: 5
test_interval: 0.25 #epoches
lr: 0.01
lr_decay_interval: 4 #epoches
lr_decay_gamma: 0.1
weight_decay: 0.00001
momentum: 0.9
model_save_path: "./params/"
checkpoint: "./checkpoint/"
data_path: "./data"
mid_layer: 512
word_embed_size: 300
token_num: 7383
max_cloth_num: 5
noise_ratio: 0.7 # 0.0, 0.1, 0.3, 0.5, 0.7
result_path: "./result"
noise_cancel_method: "forward" # forward or none
noise_loss_beta: 0.5 # the hyper-parameter for the weight of noisy samples
context: "all" # which context info to use: visual_cooc, task_ct, all, none
trans_type: "clean2noise" # noise transition matrix type: clean2noise or noise2clean
loss: "all" # which loss to use: cat, attr, all
text: 1 # whether to use text: 1 or 0
info: "final" # some log info to distinguise log file for each run