diff --git a/train.py b/train.py index 03371501..e554f424 100644 --- 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()