Laurent Mazare a27239f3d9 Add training for the llama2.c example (#296) ...

* Rework the commands and run inference by default.

* Add the training module and load the training dataset.

* Random dataset iterator.

* Proper valid-loss computation.

* Compute the evaluation loss.

* Add more substance to the training loop.

2023-08-01 17:23:07 +01:00

..

examples

Add training for the llama2.c example (#296)

2023-08-01 17:23:07 +01:00

src

Add some 'cuda-if-available' helper function. (#172)

2023-07-15 08:25:15 +01:00

build.rs

Add flash attention (#241)

2023-07-26 07:48:10 +01:00

Cargo.toml

Add training for the llama2.c example (#296)

2023-08-01 17:23:07 +01:00