diff options
| author | Patrick von Platen <patrick.v.platen@gmail.com> | 2023-08-23 10:29:50 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-08-23 09:29:50 +0100 |
| commit | 3743bed2d7bc02069770902e4a956aeabaef5453 (patch) | |
| tree | 9c65de5294df407e9146a2d169481d37006137fb /README.md | |
| parent | 508d34daf291589058929df8d91565cfce9c4375 (diff) | |
| download | candle-3743bed2d7bc02069770902e4a956aeabaef5453.tar.gz candle-3743bed2d7bc02069770902e4a956aeabaef5453.tar.bz2 candle-3743bed2d7bc02069770902e4a956aeabaef5453.zip | |
Fix the `?` operator cannot be applied to type `Device` of example (#560)
According to the API:
```rust
inp = inp.to_device(&Device::Cuda(0)?)?;
```
cannot work as `Cuda(...)` expects a type `Device` not an integer.
I'd recommend to instead use `new_cuda(...)`
Diffstat (limited to 'README.md')
| -rw-r--r-- | README.md | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -107,7 +107,7 @@ Cheatsheet: | Operations | `tensor.view((2, 2))` | `tensor.reshape((2, 2))?` | | Operations | `a.matmul(b)` | `a.matmul(&b)?` | | Arithmetic | `a + b` | `&a + &b` | -| Device | `tensor.to(device="cuda")` | `tensor.to_device(&Device::Cuda(0))?` | +| Device | `tensor.to(device="cuda")` | `tensor.to_device(&Device::new_cuda(0)?)?` | | Dtype | `tensor.to(dtype=torch.float16)` | `tensor.to_dtype(&DType::F16)?` | | Saving | `torch.save({"A": A}, "model.bin")` | `candle::safetensors::save(&HashMap::from([("A", A)]), "model.safetensors")?` | | Loading | `weights = torch.load("model.bin")` | `candle::safetensors::load("model.safetensors", &device)` | |