【Pytorch・セグメンテーション】DataParallelの読み込みについて

Question

前提・実現したいこと

現在Pytorchでのセグメンテーションを勉強しています。

学習モデルはFusionNetと呼ばれる下記のものを使用しています。
https://github.com/GunhoChoi/FusionNet-Pytorch

学習まではおそらくうまくいっていて、モデルも保存されているのですがそれを読み込んで推論（テスト）を行うことができずに困っています。
Github上には推論のプログラムまでは用意されていなかったので、見よう見まねで自作しようと思っていたのですが、モデルを読み込むところからつまずいてしまいました。

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

Traceback (most recent call last):
File "visu.py", line 17, in <module>
model.load_state_dict(state_dict)
File "C:\Users\Anaconda3\envs\FusionNet\lib\site-packages\torch\nn\modules\module.py", line 803, in load_state_dict
state_dict = state_dict.copy()
File "C:\Users\Anaconda3\envs\FusionNet\lib\site-packages\torch\nn\modules\module.py", line 576, in getattr
type(self).name, name))
AttributeError: 'DataParallel' object has no attribute 'copy'

該当のソースコード

model = nn.DataParallel(FusionGenerator(3,3,16)).cuda()
state_dict = torch.load("./model/fusion.pkl",
                        map_location={'cuda:0': 'cpu'})
model.load_state_dict(state_dict)

試したこと

とりあえず一般的なやり方として、使用したモデルを定義してtorch.loadしようとしましたが、うまくいきませんでした。

DataParallelの構造について理解しようとしましたがなかなか難しく、その読み込み方について教えていただければ幸いです。

補足情報（FW/ツールのバージョンなど）

python 3.5.2

Comments

[jbpb0]

エラーは省略しないで全部書いてください
「Traceback」と書かれてたら、そこから全部

[Gonbe_k]

編集依頼ありがとうございました。不慣れで申し訳ありません。
エラーメッセージについて全文追記いたしました。visu.pyは当該のソースコードの名前です。

[jbpb0]

model.load_state_dict(state_dict)
↓ 変更
model.load_state_dict(state_dict.module.state_dict())

で、どうでしょうか？

参考
https://discuss.pytorch.org/t/attributeerror-dataparallel-object-has-no-attribute-copy/73545/12

[jbpb0]

ダメでしょうか？