use const

src/main.rs

@@ -146,7 +146,7 @@ fn training() {
 	// training data
 	let training_samples = timelines.input_events.len() - 1;
 
-	let input_size = (size.x * size.y) as usize * size_of::<f32>();
+	let input_size = INPUT * size_of::<f32>();
 	let mut inputs = Vec::with_capacity(input_size * training_samples);
 	let mut targets = Vec::with_capacity(OUTPUT * training_samples);
 
@@ -307,10 +307,7 @@ fn training() {
 	let mut optim = AdamConfig::new().init();
 
 	let inputs = Tensor::from_data(
-		TensorData::new(
-			inputs,
-			Shape::new([training_samples, (size.x * size.y) as usize]),
-		),
+		TensorData::new(inputs, Shape::new([training_samples, INPUT])),
 		&device,
 	);
 	let targets = Tensor::from_data(