EN

Neural Network Libraries v1.19.0をリリースしました!

2021年4月15日 木曜日

リリース

Posted by Takuya Yashima

Neural Network Libraries v1.19.0をリリースしました!

Spotlight

メモリ使用量の最適化

任意のネットワークの学習過程において、メモリ使用量を削減するために以下の最適化を行いました。
– 既存の全ての関数において、Backward時に利用しない入力データをForward計算後に削除するように改善
– Backward開始時にグラフエンジンが自動で作成する勾配データ初期値を使用後に即座に削除するように改善
– メモリの共有判定(inplace)をデータと勾配データで別々に行うようにし、Backward中によりアグレッシブに勾配データを削除するように改善

参考: 1, 2, 3

線形関数の勾配計算の修正

v1.18.0で追加された高階微分の計算機能ですが、実装にミスがあり、所望の挙動となっていなかったことが判明しました。本バージョンにおいてその修正がなされています。

nnpグラフの読み込み時の問題の修正

学習済みモデルの利用時など、nnpファイルの読み込みに失敗する問題が確認されており、本バージョンでその修正を行いました。

JSI-GAN

JSI-GANのNNabla実装を追加しました!JSI-GANは超解像(Super Resolution)と逆トーンマッピング(Inverse Tone Mapping)を併用することで、低解像度・ダイナミックレンジの小さい画像を高解像度・ハイダイナミックレンジの画像に変換することができるモデルとなっています。画像の再構成/細部の復元/局所的なコントラスト増強を行うネットワークの3つを一緒に学習させることでこのタスクを実現しています。

バグ修正

ビルド関連

ドキュメントの追加・修正

フォーマットコンバーターの機能改善

レイヤー関数等の新規追加・改善

ユーティリティの機能改善

Exampleの新規追加