burn-train-0.19.1
- burn-train 0.19.1
- Docs.rs crate page
- MIT OR Apache-2.0
- Links
- Repository
- crates.io
- Source
- Owners
- nathanielsimard
- Dependencies
- async-channel ^2.5 normal
  burn-collective ^0.19.1 normal optional
  burn-core ^0.19.1 normal
  burn-ndarray ^0.19.1 normal
  burn-optim ^0.19.1 normal
  derive-new ^0.7.0 normal
  log ^0.4.28 normal
  nvml-wrapper ^0.11.0 normal optional
  ratatui ^0.29.0 normal optional
  rstest ^0.26.1 normal
  serde ^1.0.228 normal
  sysinfo ^0.37.2 normal optional
  systemstat ^0.2.5 normal optional
  tracing-appender ^0.2.3 normal
  tracing-core ^0.1.34 normal
  tracing-subscriber ^0.3.20 normal
  burn-ndarray ^0.19.1 dev
- Versions
- 100% of the crate is documented
Platform
Feature flags

docs.rs

Rust

Module train

burn_train0.19.1

Module train

Module Items

Structs

In crate burn_train

Module train

Expand description

The trainer module.

Structs§

MultiDevicesTrainStep: Multi devices train step.