metadata
datasets:
- OOPPEENN/Galgame_Dataset
language:
- ja
StableTTSの日本語事前学習
モデルパラメータ
9M
Total parameters: 9,371,649
Trainable parameters: 9,371,649
学習時のモデルパラメータ
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
トレーニングパラメータ
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