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

modify unresolved import `regression` (#2285)

This commit is contained in:
tiruka 2024-09-17 21:33:09 +09:00 committed by GitHub
parent 7ac5deebe2
commit c0656b5f9f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
examples/simple-regression/examples/regression.rs
View File

@ -9,7 +9,7 @@ mod ndarray {
ndarray::{NdArray, NdArrayDevice},
Autodiff,
};
use regression::training;
use simple_regression::training;
pub fn run() {
let device = NdArrayDevice::Cpu;
@ -23,7 +23,7 @@ mod tch_gpu {
libtorch::{LibTorch, LibTorchDevice},
Autodiff,
};
use regression::training;
use simple_regression::training;
pub fn run() {
#[cfg(not(target_os = "macos"))]
@ -41,7 +41,7 @@ mod wgpu {
wgpu::{Wgpu, WgpuDevice},
Autodiff,
};
use regression::training;
use simple_regression::training;
pub fn run() {
let device = WgpuDevice::default();
@ -55,7 +55,7 @@ mod tch_cpu {
libtorch::{LibTorch, LibTorchDevice},
Autodiff,
};
use regression::training;
use simple_regression::training;
pub fn run() {
let device = LibTorchDevice::Cpu;
training::run::<Autodiff<LibTorch>>(device);