Skip to main content

entrenar/config/train/
demo.rs

1//! Demo batch creation for testing
2
3use crate::train::Batch;
4use crate::Tensor;
5
6/// Create demo batches for testing when no data file is available
7pub fn create_demo_batches(batch_size: usize) -> Vec<Batch> {
8    let num_batches = 2.max(8 / batch_size.max(1));
9    (0..num_batches)
10        .map(|i| {
11            let input_data: Vec<f32> =
12                (0..batch_size * 4).map(|j| ((i * batch_size + j) as f32) * 0.1).collect();
13            let target_data: Vec<f32> =
14                (0..batch_size * 4).map(|j| ((i * batch_size + j + 1) as f32) * 0.1).collect();
15            Batch::new(Tensor::from_vec(input_data, false), Tensor::from_vec(target_data, false))
16        })
17        .collect()
18}