nnsvs.train_util
Specific utilities for training
Helper for training
Setup for training |
|
Save a checkpoint. |
|
Count the number of trainable parameters in the model. |
DataLoader
Yield mini-batches of indices bucketed by size. |
|
Create batch |
|
Collate function with random segments |
|
Get data loaders for training and validation. |
Plotting
Plot acoustic parameters of parametric SVS |
Misc
Compute note segments (start and end indices) from log-F0 |
|
Compute pitch regularization weight given note segments |
|
Batch version of computing pitch regularization weight |
|
Compute distortion measures between predicted and ground-truth acoustic features |
|
Ensure that the number of frames is divisible by N. |