Skip to content

Commit 0b2890c

Browse files
committed
readme: Add pretrained
1 parent e506208 commit 0b2890c

1 file changed

Lines changed: 12 additions & 0 deletions

File tree

README.md

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,18 @@ To inference test set, run [inference.py](./inference.py).
5151
python .\inference.py
5252
```
5353

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+
5466
## Learning Curve
5567

5668
res.channels=64, T=20, train 500k steps.

0 commit comments

Comments
 (0)