Commit Graph

5 Commits

Author SHA1 Message Date
chenyu b886d250fb
improve test_dropout_on_shard (#4912)
tested some basic property, also minor formatting for a few Tensor.training setups
2024-06-11 11:36:02 -04:00
George Hotz 17faae091b
optimizer shouldn't be run without training (#4460)
* optimizer shouldn't be run without training

* set training in relevant tests

* fix multitensor

* that too
2024-05-06 15:34:12 -07:00
George Hotz 3527c5a9d2
add Tensor.replace (#3738)
* add Tensor.replace

* fix dtypes in that test

* should be replace

* and mixtral
2024-03-14 13:34:14 -07:00
George Hotz 838afbc351
assign tests (#3728) 2024-03-13 17:04:55 -07:00
George Hotz ee83505fcc
fix test extra issue (#3159) 2024-01-17 11:58:08 -08:00