超解像手法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'
https://ja.stackoverflow.com/questions/51136/importerror-cannot-import-name-worker-manager-loop
にて解決しているようですので「解決済」としてください
回答1件
あなたの回答
tips
プレビュー