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