PyTorch出现：RuntimeError: An attempt has been made to start a new process...报错解决

问题：

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. 代码在运行 epoch 之前，加上 if __name__=='__main__'

if __name__ =="__main__":
    for epoch in range(1, opt.nEpochs + 1):
        train(epoch)
        test()
        checkpoint(epoch)

2. 不使用多线程，即去掉 num_workers 参数，或设置 num_workers=0（未尝试）

附：

if __name__ == '__main__' 就相当于是 Python 模拟的程序入口，如果模块是被直接运行的，则代码块被运行，如果模块是被导入的，则代码块不被运行。

参考：1. PyTorch出现：RuntimeError: An attempt has been made to start a new process.

2. Python 中的 if __name__ == '__main__' 该如何理解