def define_name(dir_name, args):
if 'conv' in args.enc_type:
dir_name = ConvEncoder.define_name(dir_name, args)
dir_name += str(args.transformer_enc_d_model) + 'dmodel'
dir_name += str(args.transformer_enc_d_ff) + 'dff'
if args.transformer_ffn_bottleneck_dim > 0:
dir_name += str(args.transformer_ffn_bottleneck_dim) + 'bn'
dir_name += str(args.enc_n_layers) + 'L'
dir_name += str(args.transformer_enc_n_heads) + 'H'
dir_name += 'pe' + str(args.transformer_enc_pe_type)
if args.transformer_enc_clamp_len > 0:
dir_name += '_clamp' + str(args.transformer_enc_clamp_len)
if args.dropout_enc_layer > 0:
dir_name += '_LD' + str(args.dropout_enc_layer)
if int(str(args.lc_chunk_size_left).split('_')[-1]) > 0 or \
int(str(args.lc_chunk_size_current).split('_')[-1]) > 0 or \
int(str(args.lc_chunk_size_right).split('_')[-1]) > 0:
dir_name += '_chunkL' + str(args.lc_chunk_size_left) + 'C' + \
str(args.lc_chunk_size_current) + 'R' + str(args.lc_chunk_size_right)
dir_name += '_' + args.lc_type
elif sum(list(map(int,
args.transformer_enc_lookaheads.split('_')))) > 0:
dir_name += '_LA' + str(
sum(list(map(int,
args.transformer_enc_lookaheads.split('_')))))
return dir_name
def define_name(dir_name, args):
if 'conv' in args.enc_type:
dir_name = ConvEncoder.define_name(dir_name, args)
dir_name += str(args.enc_n_units) + 'H'
if args.enc_n_projs > 0:
dir_name += str(args.enc_n_projs) + 'P'
dir_name += str(args.enc_n_layers) + 'L'
if args.bidirectional_sum_fwd_bwd:
dir_name += '_sumfwdbwd'
if int(args.lc_chunk_size_left.split('_')[0]) > 0 or int(args.lc_chunk_size_right.split('_')[0]) > 0:
dir_name += '_chunkL' + args.lc_chunk_size_left + 'R' + args.lc_chunk_size_right
return dir_name
def define_name(dir_name, args):
if 'conv' in args.enc_type:
dir_name = ConvEncoder.define_name(dir_name, args)
dir_name += str(args.transformer_d_model) + 'dmodel'
dir_name += str(args.transformer_d_ff) + 'dff'
if args.transformer_ffn_bottleneck_dim > 0:
dir_name += str(args.transformer_ffn_bottleneck_dim) + 'bn'
dir_name += str(args.enc_n_layers) + 'L'
dir_name += str(args.transformer_n_heads) + 'H'
dir_name += 'pe' + str(args.transformer_enc_pe_type)
if args.dropout_enc_layer > 0:
dir_name += 'droplayer' + str(args.dropout_enc_layer)
if args.lc_chunk_size_left > 0 or getattr(args, 'lc_chunk_size_current', 0) > 0 or args.lc_chunk_size_right > 0:
dir_name += '_chunkL' + str(args.lc_chunk_size_left) + 'C' + \
str(args.lc_chunk_size_current) + 'R' + str(args.lc_chunk_size_right)
dir_name += '_' + args.lc_type
return dir_name
def define_name(dir_name, args):
if 'conv' in args.enc_type:
dir_name = ConvEncoder.define_name(dir_name, args)
dir_name += str(args.enc_n_units) + 'H'
if args.enc_n_projs > 0:
dir_name += str(args.enc_n_projs) + 'P'
dir_name += str(args.enc_n_layers) + 'L'
if args.bidirectional_sum_fwd_bwd:
dir_name += '_sumfwdbwd'
if int(args.lc_chunk_size_left.split('_')[0]) > 0 or int(
args.lc_chunk_size_right.split('_')[0]) > 0:
dir_name += '_chunkL' + args.lc_chunk_size_left + 'R' + args.lc_chunk_size_right
if not args.cnn_lookahead:
dir_name += '_blockwise'
if args.rsp_prob_enc > 0:
dir_name += '_RSP' + str(args.rsp_prob_enc)
return dir_name
