Add manual seed

2018-05-11 03:47:30 -07:00 · 2018-05-11 03:47:30 -07:00 · d4bfb0e31c
parent 086160d430
commit d4bfb0e31c
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -28,7 +28,7 @@ from datasets.LJSpeech import LJSpeechDataset
 from models.tacotron import Tacotron
 from layers.losses import L1LossMasked

-
+torch.manual_seed(1)
 use_cuda = torch.cuda.is_available()

 parser = argparse.ArgumentParser()