首页 > 其他 > 详细

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

时间:2020-01-05 21:43:46      阅读:827      评论:0      收藏:0      [点我收藏+]

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.

 

dataloaders = {x: torch.utils.data.DataLoader(image_datasets[x], batch_size=opt.batchsize,
shuffle=False, num_workers=0) for x in
原因:多进程需要在main函数中运行,

解决方法1:

加main函数,在main中调用

解决方法2:

num_workers改为0,单进程加载

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

原文:https://www.cnblogs.com/blogwangwang/p/12153569.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!