pytorch-Dataloader多进程使用出错

使用Dataloader进行多进程数据导入训练时,会因为多进程的问题而出错

dataloader = DataLoader(transformed_dataset, batch_size=4,shuffle=True, num_workers=4)

其中参数num_works=表示载入数据时使用的进程数,此时如果参数的值不为0而使用多进程时会出现报错

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.

此时在数据的调用之前加上if __name__ == '__main__':即可解决问题

if __name__ == '__main__':#这个地方可以解决多线程的问题

        for i_batch, sample_batched in enumerate(dataloader):

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • rljs by sennchi Timeline of History Part One The Cognitiv...
    sennchi阅读 7,451评论 0 10
  • 不得不感叹一下,这次训练营里真是藏龙卧虎,我每天都被惊到,原来此人这么厉害。前两天有的沾沾自喜,这下真的是不敢再有...
    Peter_333e阅读 675评论 0 2
  • 空荡荡的教室 我小心翼翼地推门而进 声控灯一下子灭了 门外的声音那么清晰就在耳边 窗外的行人那么真实就在眼前 可这...
    黄粱壹梦二十年阅读 552评论 0 0
  • 贪, 貌似与钱沾边。 贪污受贿,贪得无厌! 婪, 貌似与女人有关。 实为贪吃,与性别何干? 贪婪, 看似诱惑在先—...
    觉海贝影阅读 437评论 4 8