History

mepatrick73 f7639bd35a Repeat operation (#2090 ) * renaming repeat to repeat_dim * implementing repeat function * renaming repeat files to repeat_dim * renaming part 2 * renaming part 3 * renaming part 4 * renaming part 5 * adding test file * adding unit test * adding rust book documentation * adding function args doc * fixing tests * changing repeat api to match pytorch equivalent * fixing clippy error		2024-08-02 20:33:47 -04:00
..
examples	[Breaking] Refactor Backend Names (#904 )	2023-10-29 18:27:49 -04:00
src	Repeat operation (#2090 )	2024-08-02 20:33:47 -04:00
Cargo.toml	Combined PRs (#1708 )	2024-04-29 07:52:56 -04: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