diff options
| author | Katarzyna KaĆska <kasia0308@gmail.com> | 2017-07-22 21:12:19 +0200 |
|---|---|---|
| committer | Piotr Kozakowski <kozak000@gmail.com> | 2017-07-22 19:14:33 +0000 |
| commit | 260da93a5c250e3231f59a4ce2747e2d6e215ada (patch) | |
| tree | 711c2b20dc30908bebc74d5758a761a97b2cbaeb | |
| parent | 741ef7374f15a13fe2821cc654839ed01f8037c6 (diff) | |
Delete sample.py
| -rw-r--r-- | sample.py | 34 |
1 files changed, 0 insertions, 34 deletions
diff --git a/sample.py b/sample.py deleted file mode 100644 index 775fd24..0000000 --- a/sample.py +++ /dev/null @@ -1,34 +0,0 @@ -from model import SampleRNN, Predictor, Generator -from trainer import Trainer, sequence_nll_loss -from dataset import FolderDataset, DataLoader - -import torch -from torch.utils.trainer import plugins - -from librosa.output import write_wav - -from time import time - - -def main(): - model = SampleRNN( - frame_sizes=[16, 4], n_rnn=1, dim=1024, learn_h0=True, q_levels=256 - ) - predictor = Predictor(model).cuda() - predictor.load_state_dict(torch.load('model.tar')) - - generator = Generator(predictor.model, cuda=True) - - t = time() - samples = generator(5, 16000) - print('generated in {}s'.format(time() - t)) - - write_wav( - 'sample.wav', - samples.cpu().float().numpy()[0, :], - sr=16000, - norm=True - ) - -if __name__ == '__main__': - main() |
