86 lines
1.3 KiB
INI
86 lines
1.3 KiB
INI
|
|
[DEFAULTS]
|
|
|
|
#name of the run
|
|
exp_name = diffrhythm-test
|
|
|
|
# the batch size
|
|
batch_size = 8
|
|
|
|
# the chunk size
|
|
max_frames = 3000
|
|
min_frames = 10
|
|
|
|
# number of CPU workers for the DataLoader
|
|
num_workers = 4
|
|
|
|
# the random seed
|
|
seed = 42
|
|
|
|
# Batches for gradient accumulation
|
|
accum_batches = 1
|
|
|
|
# Number of steps between checkpoints
|
|
checkpoint_every = 10000
|
|
|
|
# trainer checkpoint file to restart training from
|
|
ckpt_path = ''
|
|
|
|
# model checkpoint file to start a new training run from
|
|
pretrained_ckpt_path = ''
|
|
|
|
# Checkpoint path for the pretransform model if needed
|
|
pretransform_ckpt_path = ''
|
|
|
|
# configuration model specifying model hyperparameters
|
|
model_config = ''
|
|
|
|
# configuration for datasets
|
|
dataset_config = ''
|
|
|
|
# directory to save the checkpoints in
|
|
save_dir = ''
|
|
|
|
# grad norm
|
|
max_grad_norm = 1.0
|
|
|
|
# grad accu
|
|
grad_accumulation_steps = 1
|
|
|
|
# lr 1e-5
|
|
learning_rate = 7.5e-5
|
|
|
|
# epoch
|
|
epochs = 110
|
|
|
|
# warmup steps
|
|
num_warmup_updates = 20
|
|
|
|
# save checkpoint per steps
|
|
save_per_updates = 5000
|
|
|
|
# save last checkpoint per steps
|
|
last_per_steps = 5000
|
|
|
|
file_path = "dataset/train.scp"
|
|
|
|
audio_drop_prob = 0.3
|
|
|
|
cond_drop_prob = 0.0
|
|
|
|
style_drop_prob = 0.1
|
|
|
|
lrc_drop_prob = 0.1
|
|
|
|
use_style_prompt = 1
|
|
|
|
reset_lr = 0
|
|
|
|
resumable_with_seed = 666
|
|
|
|
downsample_rate = 2048
|
|
|
|
grad_ckpt = 0
|
|
|
|
sampling_rate = 44100
|