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 |