From 5341bf4cd51bc4744a707fd2ebcb46b851c86f18 Mon Sep 17 00:00:00 2001 From: laurent Date: Thu, 3 Apr 2025 19:30:20 +0200 Subject: [PATCH] Fixes for clippy 1.86. --- candle-nn/src/loss.rs | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/candle-nn/src/loss.rs b/candle-nn/src/loss.rs index 03e8524d..7fc349fa 100644 --- a/candle-nn/src/loss.rs +++ b/candle-nn/src/loss.rs @@ -7,7 +7,7 @@ use candle::{Result, Tensor}; /// Arguments /// /// * [inp]: The input tensor of dimensions `N, C` where `N` is the batch size and `C` the number -/// of categories. This is expected to contain log probabilities. +/// of categories. This is expected to contain log probabilities. /// * [target]: The ground truth labels as a tensor of u32 of dimension `N`. /// /// The resulting tensor is a scalar containing the average value over the batch. @@ -34,7 +34,7 @@ pub fn nll(inp: &Tensor, target: &Tensor) -> Result { /// Arguments /// /// * [inp]: The input tensor of dimensions `N, C` where `N` is the batch size and `C` the number -/// of categories. This is expected to raw logits. +/// of categories. This is expected to raw logits. /// * [target]: The ground truth labels as a tensor of u32 of dimension `N`. /// /// The resulting tensor is a scalar containing the average value over the batch. @@ -56,9 +56,9 @@ pub fn mse(inp: &Tensor, target: &Tensor) -> Result { /// Arguments /// /// * [inp]: The input tensor of dimensions `N, C` where `N` is the batch size and `C` the number -/// of categories. This is expected to raw logits. +/// of categories. This is expected to raw logits. /// * [target]: The ground truth labels as a tensor of u32 of dimension `N, C` where `N` is the batch size and `C` the number -/// of categories. +/// of categories. /// /// The resulting tensor is a scalar containing the average value over the batch. pub fn binary_cross_entropy_with_logit(inp: &Tensor, target: &Tensor) -> Result {