6 commits

Author	SHA1	Message	Date
Yixiang Gao	094d3d71be	with Tensor.train() (#1935) ... * add with.train * remove the rest TODOs * fix pyflake * fix pyflake error * fix mypy	2023-09-28 18:02:31 -07:00
George Hotz	718ced296c	move state to nn/state (#1619)	2023-08-22 07:36:24 -07:00
George Hotz	37fa7e96fb	Revert "update editorconfig, enforce via CI (#1343)" (#1380) ... This reverts commit da2efecbe2.	2023-07-31 10:35:50 -07:00
Pavol Rusnak	da2efecbe2	update editorconfig, enforce via CI (#1343) ... * update editorconfig to set unix-style newlines and trim whitespace * add editorconfig github action to the CI * fix whitespace	2023-07-30 18:44:30 -07:00
Reza Rezvan	8ae9a054ae	Refactor nn.optim (#1091) ... * Refactor: nn.optim.py * Refactor: nn.optim.py; Fix all tests * Refactor: Replace all optim.get_parameters() * Refactor: Revert list comp. * Refactor: Replace optim.get_state_dict * Refactor: Change quickstart.md	2023-07-02 15:07:30 -07:00
George Hotz	8919ca8163	test cleanups	2023-03-03 06:36:06 -08:00

Renamed from test/graph_batchnorm.py (Browse further)