search

tsubasa-alife / DeZeroUnity

Unity向けに Pure C# で実装した Neural Network 構築用フレームワーク
MIT License
0 stars 0 forks source link

学習した重みの保存機能を作る #2

Closed tsubasa-alife closed 10 months ago

tsubasa-alife commented 1 year ago

概要

学習後の重みやモデルの保存機能がないので作る必要がある。

備考

重みの保存方法を何にするのかPytorch等を参考に検討する。

tsubasa-alife commented 11 months ago

Pytorchの実装をみたところ

の2種類がある様子。

tsubasa-alife commented 10 months ago

下記コミットでモデルのパラメタ部分を保存する機能を実装。 https://github.com/tsubasa-alife/DeZeroUnity/commit/0f1ec52cee04710d479bc613ff69737d0dec8602