diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2024-03-16 22:25:46 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-03-16 22:25:46 +0100 |
| commit | 74bf6994b172f364c6e8bea2ac6e1bfbc6ca0c25 (patch) | |
| tree | f52c2638056bffa3f4d96693904ab6f58862e9cc /candle-examples/examples/efficientnet | |
| parent | cdc4c172c42b5c31b3063afd20cc7055d60f9af8 (diff) | |
| download | candle-74bf6994b172f364c6e8bea2ac6e1bfbc6ca0c25.tar.gz candle-74bf6994b172f364c6e8bea2ac6e1bfbc6ca0c25.tar.bz2 candle-74bf6994b172f364c6e8bea2ac6e1bfbc6ca0c25.zip | |
Move the image tensor to the appropriate device. (#1856)
Diffstat (limited to 'candle-examples/examples/efficientnet')
| -rw-r--r-- | candle-examples/examples/efficientnet/main.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-examples/examples/efficientnet/main.rs b/candle-examples/examples/efficientnet/main.rs index 0e4a2864..a8f17cca 100644 --- a/candle-examples/examples/efficientnet/main.rs +++ b/candle-examples/examples/efficientnet/main.rs @@ -47,7 +47,7 @@ pub fn main() -> anyhow::Result<()> { let device = candle_examples::device(args.cpu)?; - let image = candle_examples::imagenet::load_image224(args.image)?; + let image = candle_examples::imagenet::load_image224(args.image)?.to_device(&device)?; println!("loaded image {image:?}"); let model_file = match args.model { |