Commit Graph

4 Commits (dd660ed99ee4eba797b31fc730ae95e032ef13e6)

Author SHA1 Message Date
mzr1996 877ea3080a [Fix] Fix wrong `--local_rank`. 2022-05-17 21:10:25 +08:00
Ma Zerun a87ff7e153
[Enhance] Use `--a-b` instead of `--a_b` in arguments. (#754) 2022-03-30 19:16:28 +08:00
Ma Zerun 44e9902979
[Refactor] Use `--gpu-id` instead of `--gpu-ids` in non-distributed multi-gpu training/testing. (#688) 2022-02-17 02:17:01 +08:00
Ma Zerun b39885d953
[Feature] Support K-fold cross-validation (#563)
* Support to use `indices` to specify which samples to evaluate.

* Add KFoldDataset wrapper

* Rename 'K' to 'num_splits' accroding to sklearn

* Add `kfold-cross-valid.py`

* Add unit tests

* Add help doc and docstring
2022-01-19 18:32:55 +08:00