RuntimeError: An attempt has been made to start a new process before the current pr

qq_30468133

于 2019-02-22 20:20:35 发布

阅读量453

点赞数

分类专栏：小知识点

本文链接：https://blog.youkuaiyun.com/qq_30468133/article/details/87885034

版权

pytorch 加载数据时报错：

RuntimeError:
An attempt has been made to start a new process before the
current process has finished its bootstrapping phase.

        This probably means that you are not using fork to start your
        child processes and you have forgotten to use the proper idiom
        in the main module:

            if __name__ == '__main__':
                freeze_support()
                ...

The "freeze_support()" line can be omitted if the program
is not going to be frozen to produce an executable.

from torch.utils.data import DataLoader
from text_loader import TextDataset

print("hello")

if __name__=="__main__": #必须加name ==maiin才不会报错，多线程问题
    dataset = TextDataset()
    train_loader = DataLoader(dataset=dataset,