The size of tensor a (512) must match the size of tensor b (2) at non-singleton dimension 1

Question

Traceback (most recent call last):
File "/media/speechlab/data/openasr/egs/aishell2/s5/../../../src/train.py", line 136, in
trainer.train()
File "/media/speechlab/data/openasr/src/trainer.py", line 152, in train
tr_loss = self.iter_one_epoch()
File "/media/speechlab/data/openasr/src/trainer.py", line 208, in iter_one_epoch
data = next(loader_iter)
File "/home/speechlab/anaconda3/envs/pytorch3.7/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 345, in next
data = self._next_data()
File "/home/speechlab/anaconda3/envs/pytorch3.7/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 856, in _next_data
return self._process_data(data)
File "/home/speechlab/anaconda3/envs/pytorch3.7/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 881, in _process_data
data.reraise()
File "/home/speechlab/anaconda3/envs/pytorch3.7/lib/python3.7/site-packages/torch/_utils.py", line 395, in reraise
raise self.exc_type(msg)
RuntimeError: Caught RuntimeError in DataLoader worker process 3.
Original Traceback (most recent call last):
File "/home/speechlab/anaconda3/envs/pytorch3.7/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 178, in _worker_loop
data = fetcher.fetch(index)
File "/home/speechlab/anaconda3/envs/pytorch3.7/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 47, in fetch
return self.collate_fn(data)
File "/media/speechlab/data/openasr/src/data.py", line 242, in call
padded_waveforms, wave_lengths = load_wave_batch(paths)
File "/media/speechlab/data/openasr/src/data.py", line 189, in load_wave_batch
padded_waveforms[i, :lengths[i]] += waveforms[i]
RuntimeError: The size of tensor a (512) must match the size of tensor b (2) at non-singleton dimension 1

Answer 1

不要只贴报错信息，你贴代码出来呀