def check(self,
encoder_type,
decoder_type,
bidirectional=False,
attention_type='location',
subsample=False,
projection=False,
ctc_loss_weight_sub=0,
conv=False,
batch_norm=False,
residual=False,
dense_residual=False,
num_heads=1,
backward_sub=False):
print('==================================================')
print(' encoder_type: %s' % encoder_type)
print(' bidirectional: %s' % str(bidirectional))
print(' projection: %s' % str(projection))
print(' decoder_type: %s' % decoder_type)
print(' attention_type: %s' % attention_type)
print(' subsample: %s' % str(subsample))
print(' ctc_loss_weight_sub: %s' % str(ctc_loss_weight_sub))
print(' conv: %s' % str(conv))
print(' batch_norm: %s' % str(batch_norm))
print(' residual: %s' % str(residual))
print(' dense_residual: %s' % str(dense_residual))
print(' backward_sub: %s' % str(backward_sub))
print(' num_heads: %s' % str(num_heads))
print('==================================================')
if conv or encoder_type == 'cnn':
# pattern 1
# conv_channels = [32, 32]
# conv_kernel_sizes = [[41, 11], [21, 11]]
# conv_strides = [[2, 2], [2, 1]]
# poolings = [[], []]
# pattern 2 (VGG like)
conv_channels = [64, 64]
conv_kernel_sizes = [[3, 3], [3, 3]]
conv_strides = [[1, 1], [1, 1]]
poolings = [[2, 2], [2, 2]]
else:
conv_channels = []
conv_kernel_sizes = []
conv_strides = []
poolings = []
# Load batch data
splice = 1
num_stack = 1 if subsample or conv or encoder_type == 'cnn' else 2
xs, ys, ys_sub, x_lens, y_lens, y_lens_sub = generate_data(
label_type='word_char',
batch_size=2,
num_stack=num_stack,
splice=splice,
backend='chainer')
num_classes = 11
num_classes_sub = 27
# Load model
model = HierarchicalAttentionSeq2seq(
input_size=xs[0].shape[-1] // splice // num_stack, # 120
encoder_type=encoder_type,
encoder_bidirectional=bidirectional,
encoder_num_units=320,
encoder_num_proj=320 if projection else 0,
encoder_num_layers=2,
encoder_num_layers_sub=1,
attention_type=attention_type,
attention_dim=128,
decoder_type=decoder_type,
decoder_num_units=320,
decoder_num_layers=1,
decoder_num_units_sub=320,
decoder_num_layers_sub=1,
embedding_dim=64,
embedding_dim_sub=32,
dropout_input=0.1,
dropout_encoder=0.1,
dropout_decoder=0.1,
dropout_embedding=0.1,
main_loss_weight=0.8,
sub_loss_weight=0.2 if ctc_loss_weight_sub == 0 else 0,
num_classes=num_classes,
num_classes_sub=num_classes_sub,
parameter_init_distribution='uniform',
parameter_init=0.1,
recurrent_weight_orthogonal=False,
init_forget_gate_bias_with_one=True,
subsample_list=[] if not subsample else [True, False],
subsample_type='drop' if not subsample else subsample,
bridge_layer=True,
init_dec_state='first',
sharpening_factor=1,
logits_temperature=1,
sigmoid_smoothing=False,
ctc_loss_weight_sub=ctc_loss_weight_sub,
attention_conv_num_channels=10,
attention_conv_width=201,
input_channel=3,
num_stack=num_stack,
splice=splice,
conv_channels=conv_channels,
conv_kernel_sizes=conv_kernel_sizes,
conv_strides=conv_strides,
poolings=poolings,
activation='relu',
batch_norm=batch_norm,
scheduled_sampling_prob=0.1,
scheduled_sampling_max_step=200,
label_smoothing_prob=0.1,
weight_noise_std=0,
encoder_residual=residual,
encoder_dense_residual=dense_residual,
decoder_residual=residual,
decoder_dense_residual=dense_residual,
decoding_order='attend_generate_update',
bottleneck_dim=256,
bottleneck_dim_sub=256,
backward_sub=backward_sub,
num_heads=num_heads,
num_heads_sub=num_heads)
# Count total parameters
for name in sorted(list(model.num_params_dict.keys())):
num_params = model.num_params_dict[name]
print("%s %d" % (name, num_params))
print("Total %.3f M parameters" % (model.total_parameters / 1000000))
# Define optimizer
learning_rate = 1e-3
model.set_optimizer('adam',
learning_rate_init=learning_rate,
weight_decay=1e-6,
lr_schedule=False,
factor=0.1,
patience_epoch=5)
# Define learning rate controller
lr_controller = Controller(learning_rate_init=learning_rate,
backend='chainer',
decay_start_epoch=20,
decay_rate=0.9,
decay_patient_epoch=10,
lower_better=True)
# GPU setting
model.set_cuda(deterministic=False, benchmark=True)
# Train model
max_step = 300
start_time_step = time.time()
for step in range(max_step):
# Step for parameter update
loss, loss_main, loss_sub = model(xs, ys, x_lens, y_lens, ys_sub,
y_lens_sub)
model.optimizer.target.cleargrads()
model.cleargrads()
loss.backward()
loss.unchain_backward()
model.optimizer.update()
if (step + 1) % 10 == 0:
# Compute loss
loss, loss_main, loss_sub = model(xs,
ys,
x_lens,
y_lens,
ys_sub,
y_lens_sub,
is_eval=True)
# Decode
best_hyps, _, _ = model.decode(
xs,
x_lens,
beam_width=1,
# beam_width=2,
max_decode_len=30)
best_hyps_sub, _, _ = model.decode(
xs,
x_lens,
beam_width=1,
# beam_width=2,
max_decode_len=60,
task_index=1)
str_hyp = idx2word(best_hyps[0][:-1]).split('>')[0]
str_ref = idx2word(ys[0])
str_hyp_sub = idx2char(best_hyps_sub[0][:-1]).split('>')[0]
str_ref_sub = idx2char(ys_sub[0])
# Compute accuracy
try:
wer, _, _, _ = compute_wer(ref=str_ref.split('_'),
hyp=str_hyp.split('_'),
normalize=True)
cer, _, _, _ = compute_wer(
ref=list(str_ref_sub.replace('_', '')),
hyp=list(str_hyp_sub.replace('_', '')),
normalize=True)
except:
wer = 1
cer = 1
duration_step = time.time() - start_time_step
print(
'Step %d: loss=%.3f(%.3f/%.3f) / wer=%.3f / cer=%.3f / lr=%.5f (%.3f sec)'
% (step + 1, loss, loss_main, loss_sub, wer, cer,
learning_rate, duration_step))
start_time_step = time.time()
# Visualize
print('Ref: %s' % str_ref)
print('Hyp (word): %s' % str_hyp)
print('Hyp (char): %s' % str_hyp_sub)
if cer < 0.1:
print('Modle is Converged.')
break
# Update learning rate
model.optimizer, learning_rate = lr_controller.decay_lr(
optimizer=model.optimizer,
learning_rate=learning_rate,
epoch=step,
value=wer)
def check(self,
encoder_type,
bidirectional=False,
subsample=False,
projection=False,
conv=False,
batch_norm=False,
activation='relu',
encoder_residual=False,
encoder_dense_residual=False,
label_smoothing=False):
print('==================================================')
print(' encoder_type: %s' % encoder_type)
print(' bidirectional: %s' % str(bidirectional))
print(' projection: %s' % str(projection))
print(' subsample: %s' % str(subsample))
print(' conv: %s' % str(conv))
print(' batch_norm: %s' % str(batch_norm))
print(' encoder_residual: %s' % str(encoder_residual))
print(' encoder_dense_residual: %s' % str(encoder_dense_residual))
print(' label_smoothing: %s' % str(label_smoothing))
print('==================================================')
if conv or encoder_type == 'cnn':
# pattern 1
# conv_channels = [32, 32]
# conv_kernel_sizes = [[41, 11], [21, 11]]
# conv_strides = [[2, 2], [2, 1]]
# poolings = [[], []]
# pattern 2 (VGG like)
conv_channels = [64, 64]
conv_kernel_sizes = [[3, 3], [3, 3]]
conv_strides = [[1, 1], [1, 1]]
poolings = [[2, 2], [2, 2]]
fc_list = [786, 786]
else:
conv_channels = []
conv_kernel_sizes = []
conv_strides = []
poolings = []
fc_list = []
# Load batch data
num_stack = 1 if subsample or conv or encoder_type == 'cnn' else 2
splice = 1
xs, ys, ys_sub, x_lens, y_lens, y_lens_sub = generate_data(
label_type='word_char',
batch_size=2,
num_stack=num_stack,
splice=splice)
num_classes = 11
num_classes_sub = 27
# Load model
model = HierarchicalCTC(
input_size=xs.shape[-1] // splice // num_stack, # 120
encoder_type=encoder_type,
encoder_bidirectional=bidirectional,
encoder_num_units=256,
encoder_num_proj=256 if projection else 0,
encoder_num_layers=2,
encoder_num_layers_sub=1,
fc_list=fc_list,
fc_list_sub=fc_list,
dropout_input=0.1,
dropout_encoder=0.1,
main_loss_weight=0.8,
sub_loss_weight=0.2,
num_classes=num_classes,
num_classes_sub=num_classes_sub,
parameter_init_distribution='uniform',
parameter_init=0.1,
recurrent_weight_orthogonal=False,
init_forget_gate_bias_with_one=True,
subsample_list=[] if not subsample else [True, False],
num_stack=num_stack,
splice=splice,
input_channel=3,
conv_channels=conv_channels,
conv_kernel_sizes=conv_kernel_sizes,
conv_strides=conv_strides,
poolings=poolings,
batch_norm=batch_norm,
label_smoothing_prob=0.1 if label_smoothing else 0,
weight_noise_std=0,
encoder_residual=encoder_residual,
encoder_dense_residual=encoder_dense_residual)
# Count total parameters
for name in sorted(list(model.num_params_dict.keys())):
num_params = model.num_params_dict[name]
print("%s %d" % (name, num_params))
print("Total %.3f M parameters" % (model.total_parameters / 1000000))
# Define optimizer
learning_rate = 1e-3
model.set_optimizer('adam',
learning_rate_init=learning_rate,
weight_decay=1e-6,
lr_schedule=False,
factor=0.1,
patience_epoch=5)
# Define learning rate controller
lr_controller = Controller(learning_rate_init=learning_rate,
backend='pytorch',
decay_start_epoch=20,
decay_rate=0.9,
decay_patient_epoch=10,
lower_better=True)
# GPU setting
model.set_cuda(deterministic=False, benchmark=True)
# Train model
max_step = 300
start_time_step = time.time()
for step in range(max_step):
# Step for parameter update
model.optimizer.zero_grad()
loss, loss_main, loss_sub = model(xs, ys, x_lens, y_lens, ys_sub,
y_lens_sub)
loss.backward()
torch.nn.utils.clip_grad_norm_(model.parameters(), 5)
model.optimizer.step()
if (step + 1) % 10 == 0:
# Compute loss
loss, loss_main, loss_sub = model(xs,
ys,
x_lens,
y_lens,
ys_sub,
y_lens_sub,
is_eval=True)
# Decode
best_hyps, _, _ = model.decode(xs,
x_lens,
beam_width=2,
task_index=0)
best_hyps_sub, _, _ = model.decode(xs,
x_lens,
beam_width=2,
task_index=1)
str_ref = idx2word(ys[0, :y_lens[0]])
str_hyp = idx2word(best_hyps[0])
str_ref_sub = idx2char(ys_sub[0, :y_lens_sub[0]])
str_hyp_sub = idx2char(best_hyps_sub[0])
# Compute accuracy
try:
wer, _, _, _ = compute_wer(ref=str_ref.split('_'),
hyp=str_hyp.split('_'),
normalize=True)
cer, _, _, _ = compute_wer(
ref=list(str_ref_sub.replace('_', '')),
hyp=list(str_hyp_sub.replace('_', '')),
normalize=True)
except:
wer = 1
cer = 1
duration_step = time.time() - start_time_step
print(
'Step %d: loss=%.3f(%.3f/%.3f) / wer=%.3f / cer=%.3f / lr=%.5f (%.3f sec)'
% (step + 1, loss, loss_main, loss_sub, wer, cer,
learning_rate, duration_step))
start_time_step = time.time()
# Visualize
print('Ref: %s' % str_ref)
print('Hyp (word): %s' % str_hyp)
print('Hyp (char): %s' % str_hyp_sub)
if cer < 0.1:
print('Modle is Converged.')
break
# Update learning rate
model.optimizer, learning_rate = lr_controller.decay_lr(
optimizer=model.optimizer,
learning_rate=learning_rate,
epoch=step,
value=wer)
def check(self, usage_dec_sub='all', att_reg_weight=1,
main_loss_weight=0.5, ctc_loss_weight_sub=0,
dec_attend_temperature=1,
dec_sigmoid_smoothing=False,
backward_sub=False, num_heads=1, second_pass=False,
relax_context_vec_dec=False):
print('==================================================')
print(' usage_dec_sub: %s' % usage_dec_sub)
print(' att_reg_weight: %s' % str(att_reg_weight))
print(' main_loss_weight: %s' % str(main_loss_weight))
print(' ctc_loss_weight_sub: %s' % str(ctc_loss_weight_sub))
print(' dec_attend_temperature: %s' % str(dec_attend_temperature))
print(' dec_sigmoid_smoothing: %s' % str(dec_sigmoid_smoothing))
print(' backward_sub: %s' % str(backward_sub))
print(' num_heads: %s' % str(num_heads))
print(' second_pass: %s' % str(second_pass))
print(' relax_context_vec_dec: %s' % str(relax_context_vec_dec))
print('==================================================')
# Load batch data
splice = 1
num_stack = 1
xs, ys, ys_sub, x_lens, y_lens, y_lens_sub = generate_data(
label_type='word_char',
batch_size=2,
num_stack=num_stack,
splice=splice)
# Load model
model = NestedAttentionSeq2seq(
input_size=xs.shape[-1] // splice // num_stack, # 120
encoder_type='lstm',
encoder_bidirectional=True,
encoder_num_units=256,
encoder_num_proj=0,
encoder_num_layers=2,
encoder_num_layers_sub=2,
attention_type='location',
attention_dim=128,
decoder_type='lstm',
decoder_num_units=256,
decoder_num_layers=1,
decoder_num_units_sub=256,
decoder_num_layers_sub=1,
embedding_dim=64,
embedding_dim_sub=32,
dropout_input=0.1,
dropout_encoder=0.1,
dropout_decoder=0.1,
dropout_embedding=0.1,
main_loss_weight=0.8,
sub_loss_weight=0.2 if ctc_loss_weight_sub == 0 else 0,
num_classes=11,
num_classes_sub=27 if not second_pass else 11,
parameter_init_distribution='uniform',
parameter_init=0.1,
recurrent_weight_orthogonal=False,
init_forget_gate_bias_with_one=True,
subsample_list=[True, False],
subsample_type='drop',
init_dec_state='first',
sharpening_factor=1,
logits_temperature=1,
sigmoid_smoothing=False,
ctc_loss_weight_sub=ctc_loss_weight_sub,
attention_conv_num_channels=10,
attention_conv_width=201,
num_stack=num_stack,
splice=1,
conv_channels=[],
conv_kernel_sizes=[],
conv_strides=[],
poolings=[],
batch_norm=False,
scheduled_sampling_prob=0.1,
scheduled_sampling_max_step=200,
label_smoothing_prob=0.1,
weight_noise_std=0,
encoder_residual=False,
encoder_dense_residual=False,
decoder_residual=False,
decoder_dense_residual=False,
decoding_order='attend_generate_update',
# decoding_order='attend_update_generate',
# decoding_order='conditional',
bottleneck_dim=256,
bottleneck_dim_sub=256,
backward_sub=backward_sub,
num_heads=num_heads,
num_heads_sub=num_heads,
num_heads_dec=num_heads,
usage_dec_sub=usage_dec_sub,
att_reg_weight=att_reg_weight,
dec_attend_temperature=dec_attend_temperature,
dec_sigmoid_smoothing=dec_attend_temperature,
relax_context_vec_dec=relax_context_vec_dec,
dec_attention_type='location')
# Count total parameters
for name in sorted(list(model.num_params_dict.keys())):
num_params = model.num_params_dict[name]
print("%s %d" % (name, num_params))
print("Total %.3f M parameters" % (model.total_parameters / 1000000))
# Define optimizer
learning_rate = 1e-3
model.set_optimizer('adam',
learning_rate_init=learning_rate,
weight_decay=1e-6,
lr_schedule=False,
factor=0.1,
patience_epoch=5)
# Define learning rate controller
lr_controller = Controller(learning_rate_init=learning_rate,
backend='pytorch',
decay_start_epoch=20,
decay_rate=0.9,
decay_patient_epoch=10,
lower_better=True)
# GPU setting
model.set_cuda(deterministic=False, benchmark=True)
# Train model
max_step = 300
start_time_step = time.time()
for step in range(max_step):
# Step for parameter update
model.optimizer.zero_grad()
if second_pass:
loss = model(xs, ys, x_lens, y_lens)
else:
loss, loss_main, loss_sub = model(
xs, ys, x_lens, y_lens, ys_sub, y_lens_sub)
loss.backward()
torch.nn.utils.clip_grad_norm_(model.parameters(), 5)
model.optimizer.step()
if (step + 1) % 10 == 0:
# Compute loss
if second_pass:
loss = model(xs, ys, x_lens, y_lens, is_eval=True)
else:
loss, loss_main, loss_sub = model(
xs, ys, x_lens, y_lens, ys_sub, y_lens_sub, is_eval=True)
best_hyps, _, best_hyps_sub, _, perm_idx = model.decode(
xs, x_lens, beam_width=1,
max_decode_len=30,
max_decode_len_sub=60)
str_hyp = idx2word(best_hyps[0][:-1])
str_ref = idx2word(ys[0])
if second_pass:
str_hyp_sub = idx2word(best_hyps_sub[0][:-1])
str_ref_sub = idx2word(ys[0])
else:
str_hyp_sub = idx2char(best_hyps_sub[0][:-1])
str_ref_sub = idx2char(ys_sub[0])
# Compute accuracy
try:
wer, _, _, _ = compute_wer(ref=str_ref.split('_'),
hyp=str_hyp.split('_'),
normalize=True)
if second_pass:
cer, _, _, _ = compute_wer(ref=str_ref.split('_'),
hyp=str_hyp_sub.split('_'),
normalize=True)
else:
cer, _, _, _ = compute_wer(
ref=list(str_ref_sub.replace('_', '')),
hyp=list(str_hyp_sub.replace('_', '')),
normalize=True)
except:
wer = 1
cer = 1
duration_step = time.time() - start_time_step
if second_pass:
print('Step %d: loss=%.3f / wer=%.3f / cer=%.3f / lr=%.5f (%.3f sec)' %
(step + 1, loss, wer, cer, learning_rate, duration_step))
else:
print('Step %d: loss=%.3f(%.3f/%.3f) / wer=%.3f / cer=%.3f / lr=%.5f (%.3f sec)' %
(step + 1, loss, loss_main, loss_sub,
wer, cer, learning_rate, duration_step))
start_time_step = time.time()
# Visualize
print('Ref: %s' % str_ref)
print('Hyp (word): %s' % str_hyp)
print('Hyp (char): %s' % str_hyp_sub)
if cer < 0.1:
print('Modle is Converged.')
break
# Update learning rate
model.optimizer, learning_rate = lr_controller.decay_lr(
optimizer=model.optimizer,
learning_rate=learning_rate,
epoch=step,
value=wer)
