init commit

collator_args.yml

@@ -0,0 +1,16 @@
+bin_size: 128
+drop_input_prob: 0.05
+energy_max: 0.2
+energy_min: 0
+mask_length: 1
+mask_length_max: 128
+mask_proportion: 0.475
+mask_proportion_tolerance: 0.05
+max_length: 512
+name: algo
+overwrite: false
+pitch_max: 300
+pitch_min: 50
+vad_max: 1
+vad_min: 0
+vocex_model: cdminix/vocex

model_config.yml

@@ -0,0 +1,8 @@
+bins: 128
+dropout: 0.1
+filter_size: 256
+hidden_dim: 512
+kernel_size: 7
+max_length: 512
+n_heads: 2
+n_layers: 16

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7a59ced3572a2df6cd6681f1d9861f70a87e65c7bfc646f9b3335809281cd34b
+size 56214005

training_args.yml

@@ -0,0 +1,27 @@
+batch_size: 32
+checkpoint_path: checkpoints
+dataset: cdminix/libritts-aligned
+do_save: true
+dryrun: false
+eval_every_n_steps: 1000
+eval_only: false
+gradient_clip_val: 1.0
+hub_repo: null
+log_every_n_steps: 100
+lr: 0.0005
+lr_schedule: linear_with_warmup
+lr_warmup_steps: 1000
+n_epochs: 7
+n_steps: 10000
+n_workers: 96
+output_path: outputs
+push_to_hub: false
+run_name: bin128_mask_random
+save_every_n_steps: 5000
+save_onnx: false
+seed: 0
+train_split: train
+val_split: dev[:25%]
+wandb_dir: wandb
+wandb_mode: online
+wandb_project: masked-prosody-modeling-v2