diff --git a/train/process_ckpt.py b/train/process_ckpt.py index 916165e..ac608ae 100644 --- a/train/process_ckpt.py +++ b/train/process_ckpt.py @@ -17,24 +17,24 @@ def savee(ckpt, sr, if_f0, name, epoch, version, hps): continue opt["weight"][key] = ckpt[key].half() opt["config"] = [ - hpt.data.filter_length // 2 + 1, + hps.data.filter_length // 2 + 1, 32, - hpt.model.inter_channels, - hpt.model.hidden_channels, - hpt.model.filter_channels, - hpt.model.n_heads, - hpt.model.n_layers, - hpt.model.kernel_size, - hpt.model.p_dropout, - hpt.model.resblock, - hpt.model.resblock_kernel_sizes, - hpt.model.resblock_dilation_sizes, - hpt.model.upsample_rates, - hpt.model.upsample_initial_channel, - hpt.model.upsample_kernel_sizes, - hpt.model.spk_embed_dim, - hpt.model.gin_channels, - hpt.data.sampling_rate, + hps.model.inter_channels, + hps.model.hidden_channels, + hps.model.filter_channels, + hps.model.n_heads, + hps.model.n_layers, + hps.model.kernel_size, + hps.model.p_dropout, + hps.model.resblock, + hps.model.resblock_kernel_sizes, + hps.model.resblock_dilation_sizes, + hps.model.upsample_rates, + hps.model.upsample_initial_channel, + hps.model.upsample_kernel_sizes, + hps.model.spk_embed_dim, + hps.model.gin_channels, + hps.data.sampling_rate, ] opt["info"] = "%sepoch" % epoch opt["sr"] = sr