p1ch2/3_cyclegan.ipynbにつきまして

[kaz175]

はじめまして
PyTorch実践入門 Kindle版を購入して、勉強しています。

p1ch2/3_cyclegan.ipynbについて、jupyter notebook　環境で実行しています。

下記の作成済のモデルをnetGに渡すプログラムで、
　model_path = './data/p1ch2/horse2zebra_0.4.0.pth'
　model_data = torch.load(model_path)
　netG.load_state_dict(model_data)

model_data = torch.load(model_path)の段階で下記のエラーが出ます。

UnpicklingError: invalid load key, '\x0a'.

このため、データを'./data/p1ch2/horse2zebra_0.3.1.pth'に変えて実行したところ、
　model_path = './data/p1ch2/horse2zebra_0.3.1.pth'
　model_data = torch.load(model_path)
　netG.load_state_dict(model_data)

model_data = torch.load(model_path)はできましたが、netG.load_state_dict(model_data)の段階で、下記のエラーが出ます。
RuntimeError: Error(s) in loading state_dict for ResNetGenerator:

私の環境では、'horse2zebra_0.4.0.pth'は、Unpicklingできない、'horse2zebra_0.3.1.pth'は、Unpicklingできるが、netG.load_state_dict(model_data)できないため、参考までにご連絡します。

[Gin5050]

＠kaz175様

ご連絡いただきありがとうございます。

こちら本日中に確認いたしますので、＠kaz175様の実行環境について教えていただいてもよろしいでしょうか？
（PyTorchのバージョン、Pythonのバージョンなど）

先ほどColaboratoryで実行したところ問題なく動作いたしましたので、私の確認が終えるまでお手数ですが、Colabratoryで利用していただけると幸いです。

[kaz175]

Gin5050様
ご返答ありがとうございます。
Pythonのバージョンは、3.8.5、PyTorchのバージョンは1.7.1です。
当面、Colabratoryは利用いたします。

[Gin5050]

＠kaz175様

バージョン情報ありがとうございます。

いただいたバージョン情報でローカルのJupyter notebookで実行したところ、こちらの環境では問題なく動作することが確認できました。

おそらくですが、ダウンロードしたファイルが破損した可能性が考えられます。
'./data/p1ch2/horse2zebra_0.4.0.pth'のファイルを再度ダウンロードしていただき、実行していただけますでしょうか。

日本語版リポジトリのものが破損している可能性も考えられるので原著のリポジトリのリンクも貼っておきます。

https://github.com/deep-learning-with-pytorch/dlwpt-code/blob/master/data/p1ch2/horse2zebra_0.4.0.pth

どうぞよろしくお願いいたします。

[kaz175]

@Gin5050様

原著のリンクのデータに変更したところ、jupyter notebookの環境で実行できました。

本のとおり、ウマがシマウマになりました。

ありがとうございました。

[Gin5050]

＠kaz175様

解決できてよかったです！

リポジトリのデータを更新しておきます。
貴重な情報提供ありがとうございました。