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

Remove inference testing against dataset (#408) 

This test was redundant and required the dataset to be downloaded.
This commit is contained in:
Dilshod Tadjibaev 2023-06-17 11:59:26 -05:00 committed by GitHub
parent b6f88f8fca
commit 11bdceff56
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
examples/mnist-inference-web/src/web.rs
View File

@ -63,19 +63,6 @@ impl Mnist {
mod tests {
use super::Mnist;
use burn_dataset::source::huggingface::MNISTDataset;
use burn_dataset::Dataset;
#[test]
fn inference() {
let dataset = MNISTDataset::test();
let item = dataset.get(2).unwrap();
let mnist = Mnist::new();
let input = item.image.iter().copied().flatten().collect::<Vec<f32>>();
let output = mnist.inference(input.as_slice()).unwrap();
assert!(output[item.label] > 0.9);
}
#[test]
fn inference_manual_from_test_data() {