History

Guillaume Lagrange 0cbe9a927d Add learner training report summary (#1591 ) * Add training report summary * Fix LossMetric batch size state * Add NumericEntry de/serialize * Fix clippy suggestion * Compact recorder does not use compression (anymore) * Add learner summary expected results tests * Add summary to learner builder and automatically display in fit - Add LearnerSummaryConfig - Keep track of summary metrics names - Add model field when displaying from learner.fit()		2024-04-11 12:32:25 -04:00
..
examples	[Breaking] Refactor Backend Names (#904 )	2023-10-29 18:27:49 -04:00
src	Add learner training report summary (#1591 )	2024-04-11 12:32:25 -04:00
Cargo.toml	[refactor] Move burn crates to their own crates directory (#1336 )	2024-02-20 13:57:55 -05:00
README.md	Update TORCH_CUDA_VERSION usage (#1284 )	2024-02-10 12:01:45 -05:00

README.md

Text Generation

Note
This example makes use of the HuggingFace datasets library to download the datasets. Make sure you have Python installed on your computer.

The example can be run like so:

CUDA users

git clone https://github.com/tracel-ai/burn.git
cd burn

# Use the --release flag to really speed up training.
export TORCH_CUDA_VERSION=cu121
cargo run --example text-generation --release

Mac users

git clone https://github.com/tracel-ai/burn.git
cd burn

# Use the --release flag to really speed up training.
cargo run --example text-generation --release