Skip to content

Latest commit

 

History

History
 
 

text-classification

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
examples
examples
 
 
src
src
 
 
Cargo.toml
Cargo.toml
 
 
README.md
README.md
 
 

README.md

Text Classification

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                            # Set the cuda version
cargo run --example text-classification-ag-news --release  # Train on the ag news dataset
cargo run --example text-classification-db-pedia --release # Train on the db pedia dataset