dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

minor clean

This commit is contained in:
louisfd 2023-12-21 16:12:35 -05:00
parent bb18452136
commit 1fc59b0861
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
burn-core/src/nn/loss/binary_cross_entropy.rs
View File

@ -118,7 +118,7 @@ impl<B: Backend> BinaryCrossEntropyLoss<B> {
mod tests { mod tests {
use super::*; use super::*;
use crate::TestBackend; use crate::TestBackend;
use burn_tensor::{activation::sigmoid, Data, Distribution}; use burn_tensor::{activation::sigmoid, Distribution};
#[test] #[test]
fn test_binary_cross_entropy() { fn test_binary_cross_entropy() {