Commit Graph
2 Commits
Author SHA1 Message Date
Kate Yurkova 5eb83e8acd Account for num_gpus rename in docs + test configs (#2332)
* fix: num-gpus to num-devices

* fix: Update test_config1.yml

* fix: Update test_config2.yml
2023-08-12 11:09:44 +00:00
Jonáš Kulhánek a0d5307dd7 [WIP] Fix model loading collate_fn non-static (#1990)
* Fix model loading collate_fn non-static

* Fix serialization

* Add tests to reproduce the error

* Comment the change to let the tests fail

* Finish changes
2023-05-29 00:56:41 -07:00