summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorKatarzyna KaƄska <kasia0308@gmail.com>2017-07-22 21:12:19 +0200
committerPiotr Kozakowski <kozak000@gmail.com>2017-07-22 19:14:33 +0000
commit260da93a5c250e3231f59a4ce2747e2d6e215ada (patch)
tree711c2b20dc30908bebc74d5758a761a97b2cbaeb
parent741ef7374f15a13fe2821cc654839ed01f8037c6 (diff)
Delete sample.py
-rw-r--r--sample.py34
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()