Traceback (most recent call last): File "/data/ybZhang/wavenet_vocoder/egs/gaussian/../..//train.py", line 1110, in <module> ema = train_loop(device, model, data_loaders, optimizer, writer, File "/data/ybZhang/wavenet_vocoder/egs/gaussian/../..//train.py", line 833, in train_loop running_loss += __train_step(device, File "/data/ybZhang/wavenet_vocoder/egs/gaussian/../..//train.py", line 745, in __train_step y_hat, _outputs, _replicas = data_parallel_workaround(model, (x, c, g, False)) File "/data/ybZhang/wavenet_vocoder/egs/gaussian/../..//train.py", line 694, in data_parallel_workaround outputs = torch.nn.parallel.parallel_apply(replicas, inputs) File "/data/ybZhang/miniconda3/lib/python3.8/site-packages/torch/nn/parallel/parallel_apply.py", line 85, in parallel_apply output.reraise() File "/data/ybZhang/miniconda3/lib/python3.8/site-packages/torch/_utils.py", line 394, in reraise raise self.exc_type(msg) AssertionError: Caught AssertionError in replica 0 on device 0. Original Traceback (most recent call last): File "/data/ybZhang/miniconda3/lib/python3.8/site-packages/torch/nn/parallel/parallel_apply.py", line 60, in _worker output = module(*input, **kwargs) File "/data/ybZhang/miniconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 532, in __call__ result = self.forward(*input, **kwargs) File "/data/ybZhang/wavenet_vocoder/wavenet_vocoder/wavenet.py", line 197, in forward assert c.size(-1) == x.size(-1) AssertionError
Traceback (most recent call last): File "/data/ybZhang/wavenet_vocoder/egs/gaussian/../..//train.py", line 1110, in <module> ema = train_loop(device, model, data_loaders, optimizer, writer, File "/data/ybZhang/wavenet_vocoder/egs/gaussian/../..//train.py", line 833, in train_loop running_loss += __train_step(device, File "/data/ybZhang/wavenet_vocoder/egs/gaussian/../..//train.py", line 745, in __train_step y_hat, _outputs, _replicas = data_parallel_workaround(model, (x, c, g, False)) File "/data/ybZhang/wavenet_vocoder/egs/gaussian/../..//train.py", line 694, in data_parallel_workaround outputs = torch.nn.parallel.parallel_apply(replicas, inputs) File "/data/ybZhang/miniconda3/lib/python3.8/site-packages/torch/nn/parallel/parallel_apply.py", line 85, in parallel_apply output.reraise() File "/data/ybZhang/miniconda3/lib/python3.8/site-packages/torch/_utils.py", line 394, in reraise raise self.exc_type(msg) AssertionError: Caught AssertionError in replica 0 on device 0. Original Traceback (most recent call last): File "/data/ybZhang/miniconda3/lib/python3.8/site-packages/torch/nn/parallel/parallel_apply.py", line 60, in _worker output = module(*input, **kwargs) File "/data/ybZhang/miniconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 532, in __call__ result = self.forward(*input, **kwargs) File "/data/ybZhang/wavenet_vocoder/wavenet_vocoder/wavenet.py", line 197, in forward assert c.size(-1) == x.size(-1) AssertionError