2 Commits

Author SHA1 Message Date
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