fff-rs / juice

The Hacker's Machine Learning Engine
1.11k stars 76 forks source link

Now also saves bias layers #193

Closed opfromthestart closed 1 year ago

opfromthestart commented 1 year ago

What does this PR accomplish?

It saves bias layers along with ordinary weight layers, allowing models to be saved and loaded from disk.

Closes #188 .

Changes proposed by this PR:

I add another name to the list of names within a layer to account for the bias weights.

Notes to reviewer:

📜 Checklist

drahnr commented 1 year ago

Thank you!

drahnr commented 1 year ago

The only thing left would be does this work with the store/load unit test added in #190 ?