在学习李沐老师的pytorch课程,在课程09 Softmax 回归 + 损失函数 + 图片分类数据集中,P4里的代码加载数据集时出现如下错误:
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.
经查找解决方法如下:
方法1:在要运行的语句前加上 if name == ‘main’:语句,如下图:
方法2:按ctrl键进入加载数据的函数源码中,找到get_loader_worker()函数将使用进程数由4改为0