mirror of https://github.com/tracel-ai/burn.git
minor clean
This commit is contained in:
parent
bb18452136
commit
1fc59b0861
|
@ -118,7 +118,7 @@ impl<B: Backend> BinaryCrossEntropyLoss<B> {
|
|||
mod tests {
|
||||
use super::*;
|
||||
use crate::TestBackend;
|
||||
use burn_tensor::{activation::sigmoid, Data, Distribution};
|
||||
use burn_tensor::{activation::sigmoid, Distribution};
|
||||
|
||||
#[test]
|
||||
fn test_binary_cross_entropy() {
|
||||
|
|
Loading…
Reference in New Issue