search

Rayhane-mamah / Tacotron-2

DeepMind's Tacotron-2 Tensorflow implementation
MIT License
2.28k stars 905 forks source link

Errors in training WaveNet #518

Open Omnizing opened 2 years ago

Omnizing commented 2 years ago

Error occured when training WaveNet model after Tacotron model successfully trained:

Traceback (most recent call last): File "train.py", line 138, in main() File "train.py", line 130, in main wavenet_train(args, log_dir, hparams, args.wavenet_input) File "/home/zts/Tacotron-2-master/wavenet_vocoder/train.py", line 346, in wavenet_train return train(log_dir, args, hparams, input_path) File "/home/zts/Tacotron-2-master/wavenet_vocoder/train.py", line 230, in train model, stats = model_train_mode(args, feeder, hparams, global_step) File "/home/zts/Tacotron-2-master/wavenet_vocoder/train.py", line 173, in model_train_mode model = create_model(model_name or args.model, hparams, init) File "/home/zts/Tacotron-2-master/wavenet_vocoder/models/init.py", line 12, in create_model return WaveNet(hparams, init) File "/home/zts/Tacotron-2-master/wavenet_vocoder/models/wavenet.py", line 109, in init name='input_convolution') File "/home/zts/Tacotron-2-master/wavenet_vocoder/models/modules.py", line 376, in init name=name, **kwargs File "/home/zts/Tacotron-2-master/wavenet_vocoder/models/modules.py", line 230, in init self._track_checkpointable(layer, name='layer') AttributeError: 'Conv1D1x1' object has no attribute '_track_checkpointable'

And the folders below 'logs-Wavenet' is all empty, which means WaveNet model is not trained at all?