TadaoYamaoka https://blog.hatena.ne.jp/TadaoYamaoka/ TadaoYamaokaの開発日記 https://tadaoyamaoka.hatenablog.com/ dlshogi コンピュータ将棋 dlshogiのモデルの訓練に使用しているPythonスクリプトは、ベタなforループで記述しており、ミニバッチ作成部分と、ニューラルネットワークの学習の処理をシーケンシャルに実行しており並列化は行っていなかった。ミニバッチデータの作成は、盤面から入力特徴量を作成する処理が比較的重いため、C++で実装して高速化している。 それでも、ある程度CPU時間を消費している。一方ニューラルネットワークの学習処理は、GPUで処理しているため、CPUは遊んでいる状態になる。 その間に、次のミニバッチデータの作成を行えば、よりCPUの使用効率を上げることができる。PyTorchには、データローダーを並列化する… 190 <iframe src="https://hatenablog-parts.com/embed?url=https%3A%2F%2Ftadaoyamaoka.hatenablog.com%2Fentry%2F2021%2F02%2F28%2F190749" title="将棋AIの進捗 その56（データローダーの並列化） - TadaoYamaokaの開発日記" class="embed-card embed-blogcard" scrolling="no" frameborder="0" style="display: block; width: 100%; height: 190px; max-width: 500px; margin: 10px 0px;"></iframe> Hatena Blog https://hatena.blog 2021-02-28 19:07:49 rich https://tadaoyamaoka.hatenablog.com/entry/2021/02/28/190749 1.0 100%