mirror of https://github.com/tracel-ai/burn.git synced 2026-05-31 19:49:48 +09:00

Files

Charles23R 4e4c0a564c Clean learning api (#4283 )

* clean up types traits, remove paradigm trait, improve learner

* code quality

* book

* doc

* error in doc

* revert TrainStep rename and remove paradigm components

* trainstep refactor with assoc types

* fix docs

* fix warning

* comment

* remove evaluator step

2026-01-12 08:30:44 -05:00

examples

Update to edition 2024 (#2931 )

2025-03-20 10:28:51 -04:00

src

Clean learning api (#4283 )

2026-01-12 08:30:44 -05:00

Cargo.toml

Combined PRs (#4140 )

2025-12-08 07:55:54 -05:00

README.md

Bump tch to 0.22.0 (#3892 )

2025-10-20 16:22:32 -04: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=cu128
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