ayousanz's picture
Create README.md
87d7636 verified
---
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
```