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

Updated explanation to match code (#1012) 

Updated the explanation at the beginning to match the actual changes to 'Cargo.toml'. Moved down the 'edition' property to match the style of the initial 'Cargo.toml' file. Removed unnecessary comment for a self-explanatory piece of code. Some rewording.
This commit is contained in:
Maksym Hazevych 2023-11-29 16:41:07 +02:00 committed by GitHub
parent fda9e5d388
commit 87393b2070
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,14 +1,14 @@
# Model
The first step is to create a project and add the different Burn dependencies. In a `Cargo.toml`
file, add the `burn`, `burn-wgpu`, `burn-dataset`, `burn-autodiff` and `burn-train`. Note that the
`serde` dependency is necessary for serialization and is mandatory for the time being.
The first step is to create a project and add the different Burn dependencies. In the `Cargo.toml` file,
add the `burn` dependency with `train` and `wgpu` features. Note that the `serde` dependency is also mandatory
for the time being, as it is needed for serialization.
```toml
[package]
edition = "2021"
name = "my-first-burn-model"
version = "0.1.0"
edition = "2021"
[dependencies]
burn = { version = "0.10.0", features=["train", "wgpu"]}
@ -24,7 +24,6 @@ pooling and ReLU activations. We will also use dropout to improve training perfo
Let us start by creating a model in a file `model.rs`.
```rust , ignore
// Import required for the model.rs file
use burn::{
config::Config,
module::Module,