diff options
Diffstat (limited to 'candle-examples/examples/yolo-v3/darknet.rs')
-rw-r--r-- | candle-examples/examples/yolo-v3/darknet.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-examples/examples/yolo-v3/darknet.rs b/candle-examples/examples/yolo-v3/darknet.rs index 94d88d58..331e712c 100644 --- a/candle-examples/examples/yolo-v3/darknet.rs +++ b/candle-examples/examples/yolo-v3/darknet.rs @@ -13,7 +13,7 @@ struct Block { impl Block { fn get(&self, key: &str) -> Result<&str> { - match self.parameters.get(&key.to_string()) { + match self.parameters.get(key) { None => candle::bail!("cannot find {} in {}", key, self.block_type), Some(value) => Ok(value), } @@ -28,7 +28,7 @@ pub struct Darknet { impl Darknet { fn get(&self, key: &str) -> Result<&str> { - match self.parameters.get(&key.to_string()) { + match self.parameters.get(key) { None => candle::bail!("cannot find {} in net parameters", key), Some(value) => Ok(value), } |