We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e506208 commit 0b2890cCopy full SHA for 0b2890c
1 file changed
README.md
@@ -51,6 +51,18 @@ To inference test set, run [inference.py](./inference.py).
51
python .\inference.py
52
```
53
54
+Pretrained checkpoints are relased on [releases](https://github.com/revsic/tf-diffwave/releases).
55
+
56
+To use pretrained model, download files and unzip it. Followings are sample code.
57
58
+```py
59
+with open('l1.json') as f:
60
+ config = Config.load(json.load(f))
61
62
+diffwave = DiffWave(config.model)
63
+diffwave.restore('./l1_500k/l1_500000.ckpt-1').expect_partial()
64
+```
65
66
## Learning Curve
67
68
res.channels=64, T=20, train 500k steps.
0 commit comments