Q&A
アーカイブ=圧縮ファイルなので、普通に圧縮解凍ソフトで解凍できませんか？
https://sevenzip.osdn.jp/
やりたいこと
Google Colaboratoryでpytorchを使用した深層学習のモデルを学習し、
学習後の重みをtorch.saveで保存したファイルをローカルにダウンロードしたいです。
環境
Google Colaboratory
Mac
PyTorch
発生しているエラー
Colabで保存した重みファイル(weights.pth)をローカルにダウンロードする時にzip archiveファイルになってしまいます。
試したこと
Googleドライブがそういう仕様なのかと思い、
- Google Colaboratoryでpytorchを使用した深層学習のモデルを学習し、重みを
weights.pthで保存 - ローカルにダウンロードせず、そのままColabで
weights.pthの重みを読み込む
ことをしたところエラーなく重みを読み込むことができました。
python
1net_loaded = Net() #モデルのクラス 2net_loaded.load_state_dict(torch.load("./weights.pth", map_location=torch.device("cpu")))
しかし、ローカルにweights.pthをダウンロードし、同じようにすると、以下のエラーが出ます。
raise RuntimeError("{} is a zip archive (did you mean to use torch.jit.load()?)".format(f.name)) RuntimeError: weights.pth is a zip archive (did you mean to use torch.jit.load()?)
Macを使っているのですが、ダウンロードした.pthファイルの情報を見ると、「書類」というファイルになっています。
圧縮フォルダなどであれば展開すればいいということがわかるのですが、
アーカイブファイルというものはどう処理・展開すれば良いでしょうか？
宜しくお願い致します。
googleドライブから、アーカイブファイルを選択し、「アプリで開く」→「Zip, Unzip」というもので解凍はできましたが、以下のような結果になりました。
Filename Filesize
archive/data.pkl 1,011
archive/version 2
archive/data/2460613184 9,600
archive/data/2460613264 24
archive/data/2460613344 409,600
archive/data/2460613424 64
archive/data/2460613504 40
archive/data/2460613584 10,240
archive/data/2460613664 1,024
archive/data/2460613824 1,800
どのファイルも中身が文字化けで見えません(バイナリだからでしょうか)。
そもそもtorch.saveで単一のファイルになるはずが、なぜかアーカイブファイルになり、それを回答すると複数のファイルが出てきています....。
何かわかりますでしょうか？
回答1件
0
ベストアンサー
Hi, today, I take the same problem, now, I handle it.
It's very easy!
You should update your torch to new stable version (1.6.0) in your local (mac) and add a parameter in your code (let model run on your 'cpu') like this:
torch.load('/Users/***/****/model.pt',map_location=torch.device('cpu'))
then, it works!
投稿2020/08/13 15:32
編集2020/08/13 15:58
総合スコア9
Thanks! My code changed as you said, and it worked.
Please complement for Japanese.
日本語で回答を要約します。
ローカルのPC(自分の場合はMac)にインストールしているPyTorchのバージョンを1.6.0(安定バージョン)にすることで解決できました。
あなたの回答
tips
プレビュー
