dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
burn/examples/text-generation
History
Nathaniel Simard d3887bcd3d
Feat/module record (#265) 		2023-04-02 16:22:05 -04:00
..
examples feat: pre-norm transformer (#230) 2023-03-13 19:09:35 -04:00
src Feat/module record (#265) 2023-04-02 16:22:05 -04:00
Cargo.toml chore: prepare release v0.6.0 (#246) 2023-03-21 09:47:37 -04:00
README.md Feat/text generation example (#126) 2022-12-16 19:23:51 -05:00

README.md

Text Generation

The example can be run like so:

git clone https://github.com/burn-rs/burn.git
cd burn

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