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

Updated WebGPU to WGPU

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

6
project/src/neural.rs
View file

@ -6,7 +6,7 @@ mod training;
use super::config::OperationMode;
use burn::{
backend::{Autodiff, WebGpu},
backend::{Autodiff, Wgpu},
data::dataset::Dataset,
optim::AdamConfig,
};
@ -26,7 +26,7 @@ impl NeuralNetwork {
// Functions
fn train(&self) {
type MyBackend = WebGpu<f32, i32>;
type MyBackend = Wgpu<f32, i32>;
type MyAutodiffBackend = Autodiff<MyBackend>;
// Create a default Wgpu device
@ -52,7 +52,7 @@ impl NeuralNetwork {
);
}
fn infer(&self) {
type MyBackend = WebGpu<f32, i32>;
type MyBackend = Wgpu<f32, i32>;
let device = burn::backend::wgpu::WgpuDevice::default();