CameronJGrant/strafe-ai
1
Fork 0
forked from StrafesNET/strafe-ai
Code Pull Requests Activity

mega epoch

Browse Source
This commit is contained in:
Rhys Lloyd
2026-03-27 11:11:25 -07:00
parent 696a148786
commit 9248bdc362
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/main.rs
View File

@@ -325,8 +325,8 @@ fn training() {
&device, &device,
); );
const LEARNING_RATE: f64 = 0.01; const LEARNING_RATE: f64 = 0.001;
const EPOCHS: usize = 10000; const EPOCHS: usize = 100000;
for epoch in 0..EPOCHS { for epoch in 0..EPOCHS {
let predictions = model.forward(inputs.clone()); let predictions = model.forward(inputs.clone());