(ckpt, sr, if_f0, name, epoch, version, hps)
| 11 | |
| 12 | |
| 13 | def savee(ckpt, sr, if_f0, name, epoch, version, hps): |
| 14 | try: |
| 15 | opt = OrderedDict() |
| 16 | opt["weight"] = {} |
| 17 | for key in ckpt.keys(): |
| 18 | if "enc_q" in key: |
| 19 | continue |
| 20 | opt["weight"][key] = ckpt[key].half() |
| 21 | opt["config"] = [ |
| 22 | hps.data.filter_length // 2 + 1, |
| 23 | 32, |
| 24 | hps.model.inter_channels, |
| 25 | hps.model.hidden_channels, |
| 26 | hps.model.filter_channels, |
| 27 | hps.model.n_heads, |
| 28 | hps.model.n_layers, |
| 29 | hps.model.kernel_size, |
| 30 | hps.model.p_dropout, |
| 31 | hps.model.resblock, |
| 32 | hps.model.resblock_kernel_sizes, |
| 33 | hps.model.resblock_dilation_sizes, |
| 34 | hps.model.upsample_rates, |
| 35 | hps.model.upsample_initial_channel, |
| 36 | hps.model.upsample_kernel_sizes, |
| 37 | hps.model.spk_embed_dim, |
| 38 | hps.model.gin_channels, |
| 39 | hps.data.sampling_rate, |
| 40 | ] |
| 41 | opt["info"] = "%sepoch" % epoch |
| 42 | opt["sr"] = sr |
| 43 | opt["f0"] = if_f0 |
| 44 | opt["version"] = version |
| 45 | torch.save(opt, "assets/weights/%s.pth" % name) |
| 46 | return "Success." |
| 47 | except: |
| 48 | return traceback.format_exc() |
| 49 | |
| 50 | |
| 51 | def show_info(path): |
no test coverage detected