2 Commits

Author	SHA1	Message	Date
YUN, Junwoo	0b8aeddac6	Additoinal losses (#336) ... * cosine similarity loss --------- Co-authored-by: Awni Hannun <awni@apple.com> * Docstring nits	2024-01-08 14:01:13 -08:00
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