Docs.rs
burn-train-0.7.0
burn-train 0.7.0
Docs.rs crate page
MIT/Apache-2.0
Links
Repository
crates.io
Source
Owners
nathanielsimard
Dependencies
burn-core ^0.7.0
normal
derive-new ^0.5.9
normal
indicatif ^0.17.2
normal
log ^0.4.17
normal
log4rs ^1.2.0
normal
nvml-wrapper ^0.8.0
normal
rgb ^0.8.34
normal
serde ^1.0.160
normal
terminal_size ^0.2.3
normal
textplots ^0.8.0
normal
burn-ndarray ^0.7.0
dev
Versions
44.44%
of the crate is documented
Go to latest version
Platform
i686-pc-windows-msvc
i686-unknown-linux-gnu
x86_64-apple-darwin
x86_64-pc-windows-msvc
x86_64-unknown-linux-gnu
Feature flags
Rust
About docs.rs
Privacy policy
Rust website
The Book
Standard Library API Reference
Rust by Example
The Cargo Guide
Clippy Documentation
☰
Crate burn_train
Version 0.7.0
All Items
Modules
Structs
Traits
?
Crate
burn_train
source
·
[
−
]
Modules
checkpoint
logger
metric
train
Structs
AsyncTrainerCallback
ClassificationOutput
Simple classification output adapted for multiple metrics.
Learner
Learner struct encapsulating all components necessary to train a Neural Network model.
LearnerBuilder
Struct to configure and create a
learner
.
LearnerItem
MultiDevicesTrainStep
TrainEpoch
TrainOutput
ValidEpoch
Traits
LearnerCallback
TrainStep
ValidStep