OBJNULL/Dockerized-Rust-ML
Template
1
0
Fork 0
generated from OBJNULL/Dockerized-Rust
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fixed naming

Browse source
This commit is contained in:
Maddox Werts 2025-03-17 12:14:09 -04:00
parent 224ef39603
commit 48c709414c
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
project/src/neural.rs
View file

@ -43,7 +43,7 @@ impl NeuralNetwork {
); );
// Infer the model // Infer the model
inference::infer::<MyBackend>( infrence::infer::<MyBackend>(
artifact_dir, artifact_dir,
device, device,
burn::data::dataset::vision::MnistDataset::test() burn::data::dataset::vision::MnistDataset::test()