Entry point for training.
(work_dir: str, cli_args: argparse.Namespace)
| 171 | |
| 172 | |
| 173 | def main(work_dir: str, cli_args: argparse.Namespace) -> None: |
| 174 | """Entry point for training.""" |
| 175 | |
| 176 | # generate some random data for demo |
| 177 | files = make_batches( |
| 178 | n_samples_per_batch=1024, n_features=17, n_batches=31, work_dir=work_dir |
| 179 | ) |
| 180 | it = Iterator(cli_args.device, files) |
| 181 | |
| 182 | hist_train(it) |
| 183 | approx_train(it) |
| 184 | |
| 185 | |
| 186 | def setup_async_pool() -> None: |
no test coverage detected