tch-0.4.0
- tch 0.4.0
- Docs.rs crate page
- MIT/Apache-2.0
- Links
- Repository
- crates.io
- Source
- Owners
- LaurentMazare
- Dependencies
- cpython ^0.2.0 normal
  half ^1.6 normal
  lazy_static ^1.3.0 normal
  libc ^0.2.0 normal
  ndarray ^0.14 normal
  rand ^0.7 normal
  thiserror ^1 normal
  torch-sys ^0.4.0 normal
  zip ^0.5 normal
  anyhow ^1 dev
- Versions
- 40.95% of the crate is documented
Go to latest version
Platform
- i686-unknown-linux-gnu
- x86_64-unknown-linux-gnu
Feature flags

Rust

☰

Function tch::nn::sgd[−][src]

pub fn sgd(momentum: f64, dampening: f64, wd: f64, nesterov: bool) -> Sgd

Creates the configuration for a Stochastic Gradient Descent (SGD) optimizer.