Awni Hannun dff4a3833f Module checks the weight on load_weights (#337) ...

* update module to check weights on load, also fix docs and reorganize tests

* nits + rebase

* a few more docs updates for Module

* use manual module file

* comment

2024-01-02 18:55:42 -08:00

..

module-base-class.rst

Module checks the weight on load_weights (#337)

2024-01-02 18:55:42 -08:00

nn-module-template.rst

jagrit's commit files

2023-11-29 10:52:08 -08:00

optimizers-template.rst

awni's commit files

2023-11-29 10:30:41 -08:00