From d230621964fceac1783bfcb265f7140afd29350d Mon Sep 17 00:00:00 2001
From: Ivan Vovk <ivanvovk4@yandex.ru>
Date: Thu, 12 Nov 2020 12:43:10 +0300
Subject: [PATCH] forgot loading checkpoint (issue #11)

---
 inference.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/inference.py b/inference.py
index 312b276..e7ca111 100644
--- a/inference.py
+++ b/inference.py
@@ -44,6 +44,7 @@
 
     # Initialize the model
     model = WaveGrad(config)
+    model.load_state_dict(torch.load(args.checkpoint_path)['model'], strict=False)
 
     # Set noise schedule
     noise_schedule = torch.load(args.noise_schedule_path)