Errors in training WaveNet

[Omnizing]

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?