|
--- |
|
datasets: |
|
- OOPPEENN/Galgame_Dataset |
|
language: |
|
- ja |
|
--- |
|
|
|
# StableTTSの日本語事前学習 |
|
|
|
**モデルパラメータ** |
|
|
|
9M |
|
```sh |
|
Total parameters: 9,371,649 |
|
Trainable parameters: 9,371,649 |
|
``` |
|
|
|
**学習時のモデルパラメータ** |
|
|
|
```json |
|
hidden_channels: int = 192 |
|
filter_channels: int = 512 |
|
n_heads: int = 2 |
|
n_enc_layers: int = 3 |
|
n_dec_layers: int = 2 |
|
kernel_size: int = 3 |
|
p_dropout: int = 0.1 |
|
gin_channels: int = 192 |
|
``` |
|
|
|
**トレーニングパラメータ** |
|
|
|
```json |
|
train_dataset_path: str = 'filelists/filelist.json' |
|
test_dataset_path: str = 'filelists/filelist.json' |
|
batch_size: int = 48 |
|
learning_rate: float = 1e-4 |
|
num_epochs: int = 10000 |
|
model_save_path: str = './checkpoints' |
|
log_dir: str = './runs' |
|
log_interval: int = 128 |
|
save_interval: int = 15 |
|
warmup_steps: int = 200 |
|
``` |