diff options
Diffstat (limited to 'candle-examples/examples/yolo-v3/darknet.rs')
| -rw-r--r-- | candle-examples/examples/yolo-v3/darknet.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-examples/examples/yolo-v3/darknet.rs b/candle-examples/examples/yolo-v3/darknet.rs index 331e712c..1892acdd 100644 --- a/candle-examples/examples/yolo-v3/darknet.rs +++ b/candle-examples/examples/yolo-v3/darknet.rs @@ -272,7 +272,7 @@ impl Darknet { let mut prev_channels: usize = 3; for (index, block) in self.blocks.iter().enumerate() { let channels_and_bl = match block.block_type.as_str() { - "convolutional" => conv(vb.pp(&index.to_string()), index, prev_channels, block)?, + "convolutional" => conv(vb.pp(index.to_string()), index, prev_channels, block)?, "upsample" => upsample(prev_channels)?, "shortcut" => shortcut(index, prev_channels, block)?, "route" => route(index, &blocks, block)?, |