Skip to Content
ゼロから作るDeep Learning ❸ ―フレームワーク編
book

ゼロから作るDeep Learning ❸ ―フレームワーク編

by 斎藤 康毅
April 2020
Intermediate to advanced
552 pages
6h 13m
Japanese
O'Reilly Japan, Inc.
Content preview from ゼロから作るDeep Learning ❸ ―フレームワーク編

ステップ53モデルの保存と読み込み

本ステップでは、モデルが持つパラメータを外部ファイルに保存する機能を作ります。また、保存したパラメータを読み込む機能も作ります。そのような機能があれば、学習途中のモデルを「スナップショット」として保存できます。また、学習済みのパラメータを読み込んで、推論だけを行うこともできます。

さて、DeZeroのパラメータは、Parameterクラスとして実装されています(それはVariableクラスを継承したクラスです)。そしてParameterのデータは、インスタンス変数のdatandarrayインスタンスとして保持されます。そのため、ここでやるべきことは、ndarrayインスタンスを外部ファイルに保存することです。幸い、NumPyには、ndarrayの「保存」(そして「読み込み」)を行う関数がいくつか用意されています。ここでは最初に、それらの関数の使い方を見ていきます。

[注記]

DeZeroをGPUで実行する場合は、NumPyのndarrayの代わりに、CuPyのndarraycupy.ndarray)を使います。その場合は、CuPyのテンソルをNumPyのテンソルに変換してから、外部ファイルへと保存することにします。そのため、外部ファイルに保存するにあたってはNumPyの場合だけを考えます。

53.1 NumPyのsave関数とload関数

NumPyにはnp.savenp.loadという関数があります。この関数を使えば、ndarrayインスタンスの保存と読み込みが行えます。実際に試してみると、次のようになります。 ...

Become an O’Reilly member and get unlimited access to this title plus top books and audiobooks from O’Reilly and nearly 200 top publishers, thousands of courses curated by job role, 150+ live events each month,
and much more.

Read now

Unlock full access

More than 5,000 organizations count on O’Reilly

AirBnbBlueOriginElectronic ArtsHomeDepotNasdaqRakutenTata Consultancy Services

QuotationMarkO’Reilly covers everything we've got, with content to help us build a world-class technology community, upgrade the capabilities and competencies of our teams, and improve overall team performance as well as their engagement.
Julian F.
Head of Cybersecurity
QuotationMarkI wanted to learn C and C++, but it didn't click for me until I picked up an O'Reilly book. When I went on the O’Reilly platform, I was astonished to find all the books there, plus live events and sandboxes so you could play around with the technology.
Addison B.
Field Engineer
QuotationMarkI’ve been on the O’Reilly platform for more than eight years. I use a couple of learning platforms, but I'm on O'Reilly more than anybody else. When you're there, you start learning. I'm never disappointed.
Amir M.
Data Platform Tech Lead
QuotationMarkI'm always learning. So when I got on to O'Reilly, I was like a kid in a candy store. There are playlists. There are answers. There's on-demand training. It's worth its weight in gold, in terms of what it allows me to do.
Mark W.
Embedded Software Engineer

You might also like

ゼロから作るDeep Learning ❻ ―LLM編

ゼロから作るDeep Learning ❻ ―LLM編

斎藤 康毅

Publisher Resources

ISBN: 9784873119069Publisher Website