pop2piano / config.yaml
sweetcocoa's picture
initial test
88490a8
raw
history blame
1.15 kB
project: pop2piano
dataset:
target_length: 256
input_length: 1024
n_bars: 2
sample_rate: 22050
use_mel: true
mel_is_conditioned: true
composer_to_feature_token:
composer1: 2052
composer2: 2053
composer3: 2054
composer4: 2055
composer5: 2056
composer6: 2057
composer7: 2058
composer8: 2059
composer9: 2060
composer10: 2061
composer11: 2062
composer12: 2063
composer13: 2064
composer14: 2065
composer15: 2066
composer16: 2067
composer17: 2068
composer18: 2069
composer19: 2070
composer20: 2071
composer21: 2072
t5:
feed_forward_proj: gated-gelu
tie_word_embeddings: false
tie_encoder_decoder: false
vocab_size: 2400
n_positions: 1024
relative_attention_num_buckets: 32
tokenizer:
vocab_size:
special: 4
note: 128
velocity: 2
time: 100
training:
seed: 3407
resume: false
offline: false
num_gpu: 1
max_epochs: 5000
accumulate_grad_batches: 1
check_val_every_n_epoch: 20
find_lr: false
optimizer: adafactor
version: none
lr: 0.001
lr_min: 1.0e-06
lr_scheduler: false
lr_decay: 0.99
batch_size: 32
num_workers: 32
gradient_clip_val: 3.0