mirror of https://github.com/tracel-ai/burn.git
modify unresolved import `regression` (#2285)
This commit is contained in:
parent
7ac5deebe2
commit
c0656b5f9f
|
@ -9,7 +9,7 @@ mod ndarray {
|
||||||
ndarray::{NdArray, NdArrayDevice},
|
ndarray::{NdArray, NdArrayDevice},
|
||||||
Autodiff,
|
Autodiff,
|
||||||
};
|
};
|
||||||
use regression::training;
|
use simple_regression::training;
|
||||||
|
|
||||||
pub fn run() {
|
pub fn run() {
|
||||||
let device = NdArrayDevice::Cpu;
|
let device = NdArrayDevice::Cpu;
|
||||||
|
@ -23,7 +23,7 @@ mod tch_gpu {
|
||||||
libtorch::{LibTorch, LibTorchDevice},
|
libtorch::{LibTorch, LibTorchDevice},
|
||||||
Autodiff,
|
Autodiff,
|
||||||
};
|
};
|
||||||
use regression::training;
|
use simple_regression::training;
|
||||||
|
|
||||||
pub fn run() {
|
pub fn run() {
|
||||||
#[cfg(not(target_os = "macos"))]
|
#[cfg(not(target_os = "macos"))]
|
||||||
|
@ -41,7 +41,7 @@ mod wgpu {
|
||||||
wgpu::{Wgpu, WgpuDevice},
|
wgpu::{Wgpu, WgpuDevice},
|
||||||
Autodiff,
|
Autodiff,
|
||||||
};
|
};
|
||||||
use regression::training;
|
use simple_regression::training;
|
||||||
|
|
||||||
pub fn run() {
|
pub fn run() {
|
||||||
let device = WgpuDevice::default();
|
let device = WgpuDevice::default();
|
||||||
|
@ -55,7 +55,7 @@ mod tch_cpu {
|
||||||
libtorch::{LibTorch, LibTorchDevice},
|
libtorch::{LibTorch, LibTorchDevice},
|
||||||
Autodiff,
|
Autodiff,
|
||||||
};
|
};
|
||||||
use regression::training;
|
use simple_regression::training;
|
||||||
pub fn run() {
|
pub fn run() {
|
||||||
let device = LibTorchDevice::Cpu;
|
let device = LibTorchDevice::Cpu;
|
||||||
training::run::<Autodiff<LibTorch>>(device);
|
training::run::<Autodiff<LibTorch>>(device);
|
||||||
|
|
Loading…
Reference in New Issue