forked from StrafesNET/strafe-ai
tweak model
This commit is contained in:
@@ -6,9 +6,9 @@ pub type InferenceBackend = burn::backend::Cuda<f32>;
|
||||
pub type TrainingBackend = Autodiff<InferenceBackend>;
|
||||
|
||||
pub const SIZE: glam::UVec2 = glam::uvec2(64, 36);
|
||||
pub const POSITION_HISTORY: usize = 4;
|
||||
pub const POSITION_HISTORY: usize = 10;
|
||||
pub const INPUT: usize = (SIZE.x * SIZE.y) as usize + POSITION_HISTORY * 3;
|
||||
pub const HIDDEN: [usize; 2] = [INPUT >> 3, INPUT >> 7];
|
||||
pub const HIDDEN: [usize; 3] = [INPUT >> 3, INPUT >> 5, INPUT >> 7];
|
||||
// MoveForward
|
||||
// MoveLeft
|
||||
// MoveBack
|
||||
|
||||
Reference in New Issue
Block a user