diff --git a/src/main.rs b/src/main.rs index 948ed2a..fb402e1 100644 --- a/src/main.rs +++ b/src/main.rs @@ -16,7 +16,7 @@ async fn main() { println!("Done Generating MNIST"); outputs = to_categorical(outputs_uncat); - for i in 0..30000{ + for i in 0..600{ inputs.push(&inputs_undyn[i]); true_outputs.push(outputs[i].clone()); }