機械学習の学習方法について

超解像手法EDSRのトレーニング方法についての質問です．
URL：https://github.com/thstkdgus35/EDSR-PyTorch
上記のgithubにあるソースコードを用いて実行，学習を行いたいと思っています．トレーニング方法のところで，DIV2Kのデータセットをダウンロードして，DIV2Kのファイルがあるところにdir_dataの引数を変更させて行うと記述してあります．
今回，別のデータセットを用いて学習を行いたいと思っています．
option.py内のディレクトリの引数を変更を行い，以下のコードを実行しました．

python main.py --model EDSR --scale 2 --save EDSR_x2 --n_resblocks 32 --n_feats 256 --res_scale 0.1

前提・実現したいこと

上記のコードの実行を行うとエラーが返されました．
調べてみてもあてはまるものがなく手づまりです．

発生している問題・エラーメッセージ

Traceback (most recent call last):
  File "main.py", line 4, in <module>
    import data
  File "D:\EDSR-PyTorch-master\src\data\__init__.py", line 2, in <module>
    from dataloader import MSDataLoader
  File "D:\EDSR-PyTorch-master\src\dataloader.py", line 17, in <module>
    from torch.utils.data.dataloader import _worker_manager_loop
ImportError: cannot import name '_worker_manager_loop'