Skip to content

README.md

Latest commit

 

History

History
22 lines (17 loc) · 554 Bytes

README.md

File metadata and controls

22 lines (17 loc) · 554 Bytes

Sconce

Example training code

fn train_example() -> Result<()> {
    let device = &Device::Cpu;
    let variables = VarMap::new();
    let model = Sequential::new()
        .add_layer(&Dense::new(2).kernel_initializer(ZERO).build())
        .add_layer(&Activation::new(Relu).build())
        .compile(&variables, &Device::Cpu, nll, Optimizers::AdamWDefault)?;
    let x = &Tensor::new(&[1.], &device)?;
    let y = &Tensor::new(&[1.], &device)?;

    let output = model.fit(x, y, 10)?;

    println!("Loss: {}", output.loss);

    Ok(())
}