dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add vision feature for MNIST in book guide (#1267)

This commit is contained in:
Guillaume Lagrange 2024-02-07 09:21:35 -05:00 committed by GitHub
parent d2bdc46786
commit 059c241511
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
burn-book/src/basic-workflow/model.md
View File

@ -10,7 +10,7 @@ cargo new my-first-burn-model
As [mentioned previously](../getting-started.md#creating-a-burn-application), this will initialize
your `my-first-burn-model` project directory with a `Cargo.toml` and a `src/main.rs` file.
In the `Cargo.toml` file, add the `burn` dependency with `train` and `wgpu` features.
In the `Cargo.toml` file, add the `burn` dependency with `train`, `wgpu` and `vision` features.
```toml
[package]
@ -19,7 +19,7 @@ version = "0.1.0"
edition = "2021"
[dependencies]
burn = { version = "0.13.0", features = ["train", "wgpu"] }
burn = { version = "0.13.0", features = ["train", "wgpu", "vision"] }
```
Our goal will be to create a basic convolutional neural network used for image classification. We