huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 19:58:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adressed comments.

Browse Source
This commit is contained in:
Nicolas Patry
2023-08-01 14:02:21 +02:00
parent ebd0315623
commit 5cead227ef
2 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
candle-book/src/guide/hello_world.md
View File

@ -2,7 +2,7 @@
We will now create the hello world of the ML world, building a model capable of solving MNIST dataset. We will now create the hello world of the ML world, building a model capable of solving MNIST dataset.
Open `src/main.rs` and fill in with these contents: Open `src/main.rs` and fill in with this content:
```rust ```rust
# extern crate candle; # extern crate candle;
@ -45,7 +45,7 @@ cargo run --release
## Using a `Linear` layer. ## Using a `Linear` layer.
Now that we have this, we might want to complexity a little, for instance by adding `bias` and creating Now that we have this, we might want to complexify things a bit, for instance by adding `bias` and creating
the classical `Linear` layer. We can do as such the classical `Linear` layer. We can do as such
```rust ```rust
@ -76,7 +76,7 @@ impl Model {
} }
``` ```
This will change the loading code into a new function This will change the model running code into a new function
```rust ```rust
# extern crate candle; # extern crate candle;
@ -106,8 +106,10 @@ This will change the loading code into a new function
# } # }
fn main() -> Result<()> { fn main() -> Result<()> {
// Use Device::new_cuda(0)?; to use the GPU. // Use Device::new_cuda(0)?; to use the GPU.
let device = Device::Cpu; // Use Device::Cpu; to use the CPU.
let device = Device::cuda_if_available(0)?;
// Creating a dummy model
let weight = Tensor::zeros((784, 100), DType::F32, &device)?; let weight = Tensor::zeros((784, 100), DType::F32, &device)?;
let bias = Tensor::zeros((100, ), DType::F32, &device)?; let bias = Tensor::zeros((100, ), DType::F32, &device)?;
let first = Linear{weight, bias}; let first = Linear{weight, bias};
@ -118,6 +120,7 @@ fn main() -> Result<()> {
let dummy_image = Tensor::zeros((1, 784), DType::F32, &device)?; let dummy_image = Tensor::zeros((1, 784), DType::F32, &device)?;
// Inference on the model
let digit = model.forward(&dummy_image)?; let digit = model.forward(&dummy_image)?;
println!("Digit {digit:?} digit"); println!("Digit {digit:?} digit");
Ok(()) Ok(())

2
candle-book/src/guide/installation.md
View File

@ -9,7 +9,7 @@ cargo add --git https://github.com/LaurentMazare/candle.git candle
``` ```
At this point, candle will be built **without** CUDA support. At this point, candle will be built **without** CUDA support.
To get CUDA support use the feature `cuda` To get CUDA support use the `cuda` feature
```bash ```bash
cargo add --git https://github.com/LaurentMazare/candle.git candle --features cuda cargo add --git https://github.com/LaurentMazare/candle.git candle --features cuda
``` ```