Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Feat/sort FSRSItem by length to speed up training (#32)
* sort FSRSItem by length to speed up training * correct shape of logits * cargo fmt * Create batches with random seq_len * implement BatchShuffledDataset * cargo fmt * rename batch_shuffle & add English comments * Use an SQL sort and .group_by() for separating by card id * We don't need InMemDataset * Move all tests into test modules This allows easily running all tests in one file at once, and allows tests to share code that is not used in production. Also removed the duplicate test_next_stability/difficulty tests. * Remove redundant test_ prefix/suffix * Apply cosine_annealing patch from Asuka to fix test Co-authored-by: Asuka Minato <[email protected]> * Run tests in CI, except training Co-authored-by: Asuka Minato <[email protected]> * Limit checks to pull requests; bust cache --------- Co-authored-by: Damien Elmes <[email protected]> Co-authored-by: Asuka Minato <[email protected]>
- Loading branch information