visualTranslation/cfg.py at main · Bhashini-IITJ/visualTranslation · GitHub

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
36
37
38
39
40
lt = 1.
lt_alpha = 1.
lb = 1.
lb_beta = 10.
lf = 1.
lf_theta_1 = 10.
lf_theta_2 = 1.
lf_theta_3 = 500.
epsilon = 1e-8

# train
learning_rate = 1e-4
decay_rate = 0.9
beta1 = 0.9
beta2 = 0.999
max_iter = 700000
show_loss_interval = 50
write_log_interval = 50
save_ckpt_interval = 10000
gen_example_interval = 5000
num_diff_steps = 750
checkpoint_savedir = './checkpoint/'
ckpt_path = ''
batch_size = 256
data_shape = [64, None]
data_dir = 'dataset'
i_t_dir = 'i_t'
i_s_dir = 'i_s'
t_sk_dir = 't_sk'
t_t_dir = 't_t'
t_b_dir = 't_b'
t_f_dir = 't_f'
mask_t_dir = 'mask_t'
example_data_dir = 'bg_large_crops'
example_result_dir = 'gen_logs'

# predict
predict_ckpt_path = None
predict_data_dir = None
checkpoint = "model/eng_hin.model"