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

fix typo in basic-workflow training (#1000)

This commit is contained in:
Granville Schmidt 2023-11-27 07:37:29 -07:00 committed by GitHub
parent 2928657622
commit d0cb7205fa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
burn-book/src/basic-workflow/training.md
View File

@ -135,7 +135,7 @@ pub fn train<B: AutodiffBackend>(artifact_dir: &str, config: TrainingConfig, dev
It is a good practice to use the `Config` derive to create the experiment configuration. In the
`train` function, the first thing we are doing is making sure the `artifact_dir` exists, using the
standard rust library for file manipulation. All checkpoints, logging and metrics will be stored
under the this directory. We then initialize our dataloaders using our previously created batcher.
under this directory. We then initialize our dataloaders using our previously created batcher.
Since no automatic differentiation is needed during the validation phase, the backend used for the
corresponding batcher is `B::InnerBackend` (see [Backend](./backend.md)). The autodiff capabilities
are available through a type system, making it nearly impossible to forget to deactivate gradient