【EDSR代码问题】【Pytorch1.1】ImportError: cannot import name '_update_worker_pids' from 'torch._C'

原创

于 2019-05-05 16:07:37 发布 · 2.8w 阅读

74 ·

CC 4.0 BY-SA版权

文章标签：

#Pytorch1.1 #_update_worker_pids

从5月3日开始，恢复Colab上的EDSR模型训练，文件dataloader.py开始报错：

ImportError: cannot import name '_update_worker_pids'

仔细看了下原来是Colab上默认pytorch版本变成了1.1，由于没试过Colab上pytorch版本回滚，而且可预见地认为更换版本会很麻烦（每次都需要改版本），因为尝试修改EDSR代码解决该问题。

看了下pytorch1.1的GitHub源码，将EDSR中的dataloader.py需要更改处整理如下：

1

from torch._C import _set_worker_signal_handlers, _update_worker_pids, \
    _remove_worker_pids, _error_if_any_worker_fails
from torch.utils.data.dataloader import DataLoader
from torch.utils.data.dataloader import _DataLoaderIter
from torch.utils.data.dataloader import ManagerWatchdog
from torch.utils.data.dataloader import _pin_memory_loop
from torch.utils.data.dataloader import MP_STATUS_CHECK_INTERVAL

from torch.utils.data.dataloader import ExceptionWra