diff --git a/configs/48k.json b/configs/48k.json index 1f85321..981c90f 100644 --- a/configs/48k.json +++ b/configs/48k.json @@ -11,7 +11,7 @@ "eval_interval": 1000, "seed": 1234, "max_epochs": 20000, - "learning_rate": 8e-4, + "learning_rate": 4e-4, "betas": [0.8, 0.99], "eps": 1e-9, "batch_size": 32, diff --git a/hifigan/hub/__init__.py b/hifigan/hub/__init__.py index ecc71d1..d9f10bb 100644 --- a/hifigan/hub/__init__.py +++ b/hifigan/hub/__init__.py @@ -1,5 +1,5 @@ CKPT_URLS = { - "hifigan-48k": "https://github.com/vtuber-plan/hifi-gan/releases/download/v0.1.2/hifigan-48k-20DAC6A62B1508F62BA33A66CC125BBB.pt", + "hifigan-48k": "https://github.com/vtuber-plan/hifi-gan/releases/download/v0.1.3/hifigan-48k-59CB718B329ED0167F3BBD9DDC47F443.pt", } import torch from ..model.generators.generator import Generator diff --git a/save_state.py b/save_state.py index 6bb0904..3ab574f 100644 --- a/save_state.py +++ b/save_state.py @@ -21,6 +21,6 @@ def main(): ckpt_path = last_ckpt print(ckpt_path) - save("logs/lightning_logs_v1/version_8/checkpoints/last.ckpt") + save(ckpt_path) if __name__ == "__main__": main() \ No newline at end of file