-- Process 0 terminated with the following error: Traceback (most recent call last): File "/home/pai/envs/3.7/lib/python3.7/site-packages/torch/multiprocessing/spawn.py", line 69, in _wrap fn(i, *args) File "/mnt/workspace/emotional-vits/train_ms.py", line 131, in run [train_loader, eval_loader], logger, [writer, writer_eval]) File "/mnt/workspace/emotional-vits/train_ms.py", line 152, in train_and_evaluate for batch_idx, (x, x_lengths, spec, spec_lengths, y, y_lengths, speakers, emo) in enumerate(train_loader): File "/home/pai/envs/3.7/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 435, in __iter__ return self._get_iterator() File "/home/pai/envs/3.7/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 381, in _get_iterator return _MultiProcessingDataLoaderIter(self) File "/home/pai/envs/3.7/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 988, in __init__ super(_MultiProcessingDataLoaderIter, self).__init__(loader) File "/home/pai/envs/3.7/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 598, in __init__ self._sampler_iter = iter(self._index_sampler) File "/mnt/workspace/emotional-vits/data_utils.py", line 372, in __iter__ ids_bucket = ids_bucket + ids_bucket * (rem // len_bucket) + ids_bucket[:(rem % len_bucket)] ZeroDivisionError: integer division or modulo by zero
时间: 2023-08-17 13:04:05 浏览: 250
这个错误通常是由于代码中将一个数除以零或者取模运算导致的,而这个数的值为零。具体地讲,这个错误出现在了使用 PyTorch 的 DataLoader 加载数据的过程中。
根据错误提示,这个问题可能是由于数据集中的某些 batch 大小为零导致的。你可以检查你的数据集中是否存在这样的 batch,如果存在,可以将其从数据集中删除或者将其合并到其他 batch 中。
另外,你也可以检查你的代码中使用 DataLoader 的方式是否正确。例如,你可以尝试设置 `num_workers` 参数为 0,这样可以将 DataLoader 的多进程加载功能关闭,避免可能的并发问题。具体的做法是在 DataLoader 的初始化过程中添加 `num_workers=0` 参数:
```
train_loader = DataLoader(train_dataset, batch_size=batch_size, shuffle=True, num_workers=0)
```
希望这些方法能帮到你解决问题!
相关问题
torch.multiprocessing.spawn.ProcessRaisedException: -- Process 1 terminated with the following error:
根据提供的引用内容,可以看出这是一个关于PyTorch多进程训练的错误信息。具体来说,错误信息中提到了进程1以退出代码1终止,而这个错误是由于多线程程序没有放在主函数中训练所导致的。解决方法是将多线程程序放在主函数中训练。另外,也可以根据traceback的报错信息,修改代码中的mp.spawn函数的参数,例如将nprocs参数设置为1。如果还有其他问题,可以提供更详细的错误信息以便更好地解决问题。
Process 0 terminated with the following error
根据提供的引用内容,代码中的`is_broken_img`函数用于测给定路径下的图像文件是否损坏。如果图像文件的最两个字节不是JPEG文件的结束标识符`\xff\xd9`,则被认为是损坏的图像文件。该函数会输出损坏图像的文件名,并统计损坏图像的数量[^1]。
而根据第二个引用,`Process 0 terminated with the following error`错误是由于训练过程中遇到了异常文件,即不符合JPEG文件格式的图像文件。解决办法是查看异常文件,判断其是否为正确的JPEG文件。
以下是一个示例代码,用于检测给定路径下的图像文件是否损坏:
```python
import os
def is_broken_img(img_path):
num = 0
index = 0
for file_ in os.listdir(img_path):
img_ = os.path.join(img_path, file_)
with open(img_, 'rb') as f:
check_chars = f.read()[-2:]
if check_chars != b'\xff\xd9':
print("%s image is broken." % file_)
num += 1
if index % 2000 == 0:
print("index of ok image: ", index)
index += 1
print("number of broken images: ", num)
is_broken_img('/path/to/images')
```