↓ 1 callersFunctiontrain_model(epoch, model, dloader, optim, sched, pad_token, model_type="performer")
stage2_accompaniment/train.py:32
Method__init__(self, data_dir, vocab_file,
model_enc_seqlen=32, model_dec_seqlen=2400, model_max_bars=192,
stage1_compose/dataloader.py:160
Method__init__(self, pitch, bar, position, duration, velocity, microtiming=None)
stage1_compose/convert2midi.py:105
Method__init__(self, n_token, d_embed, d_proj, emb_scale=0.5, pad_idx=None)
stage1_compose/model/transformer_helpers.py:95
Method__init__(self, n_octave, d_embed, d_proj, idx2event,
emb_scale=0.5, n_chroma=12, min_pitch=12
stage1_compose/model/transformer_helpers.py:121
Method__init__(self, d_word_embed, vocab_size,
dec_n_layer, dec_n_head, dec_d_model, dec_d_ff, dec_mem_len, d
stage1_compose/model/plain_transformer.py:15
Method__init__(self, n_head, d_model, d_head, dropout, dropatt=0,
pre_lnorm=False, **kwargs)
stage1_compose/model/optimus_txl_decoder.py:65
Method__init__(self, n_head, d_model, d_head, dropout, dropatt=0,
pre_lnorm=False)
stage1_compose/model/optimus_txl_decoder.py:154
Method__init__(self, n_head, d_model, d_head, dropout, dropatt=0,
tgt_len=None, ext_len=None, mem_len=None,
stage1_compose/model/optimus_txl_decoder.py:230
Method__init__(self, n_head, d_model, d_head, d_inner, dropout,
**kwargs)
stage1_compose/model/optimus_txl_decoder.py:493
Method__init__(self, n_head, d_model, d_head, d_inner, dropout,
**kwargs)
stage1_compose/model/optimus_txl_decoder.py:512
Method__init__(self, d_in, d_out, n_layer=None, tie_seg_emb_projs=True, scale=1.)
stage1_compose/model/optimus_txl_decoder.py:560
Method__init__(self, n_layer, n_head, d_model, d_head, d_inner, d_segment_emb,
dropout, dropatt, pre_lnorm=
stage1_compose/model/optimus_txl_decoder.py:587
Method__init__(self, data_dir, vocab_file,
model_dec_seqlen=10240, model_max_bars=None,
pi
stage2_accompaniment/dataloader.py:43
Method__init__(self, pitch, bar, position, duration, velocity, microtiming=None)
stage2_accompaniment/convert2midi.py:105
Method__init__(self, n_layer, n_head, d_model, d_ff,
dropout=0.1,
activation='relu',
stage2_accompaniment/model/fast_transformer_decoder.py:13
Method__init__(self, n_token, n_layer, n_head, d_model, d_ff, d_embed,
activation='relu', dropout=0.1, use_
stage2_accompaniment/model/music_performer.py:10
Method__init__(self, n_token, n_layer, n_head, d_model, d_ff, d_embed,
activation='relu', dropout=0.1, use_pe=True,
stage2_accompaniment/model/music_gpt2.py:19
Methodforward(self, h, c, attn_mask=None, h_pos_embed=None, c_pos_embed=None)
stage1_compose/model/optimus_txl_decoder.py:85
Methodforward(self, w, r, r_w_bias, r_r_bias, attn_mask=None, mems=None, return_avg_attn=False)
stage1_compose/model/optimus_txl_decoder.py:305
Methodforward(self, w, r_emb, r_w_bias, r_bias, attn_mask=None, mems=None)
stage1_compose/model/optimus_txl_decoder.py:394
Methodforward(self, dec_inp, r_emb, r_w_bias, r_bias, dec_attn_mask=None, mems=None)
stage1_compose/model/optimus_txl_decoder.py:502
Methodforward(self, dec_inp, r, r_w_bias, r_r_bias, dec_attn_mask=None, mems=None,
cross_latent=None, dec_
stage1_compose/model/optimus_txl_decoder.py:526
Methodforward(self, dec_input, segment_emb, *mems, dec_seg_len=None, cross_latent=None,
cross_attn_mask=Non
stage1_compose/model/optimus_txl_decoder.py:927
Methodforward(self, x, seg_inp=None, chord_inp=None, keep_last_only=False, attn_kwargs=None)
stage2_accompaniment/model/music_performer.py:50