forked from StrafesNET/strafe-ai
mega epoch
This commit is contained in:
@@ -325,8 +325,8 @@ fn training() {
|
||||
&device,
|
||||
);
|
||||
|
||||
const LEARNING_RATE: f64 = 0.01;
|
||||
const EPOCHS: usize = 10000;
|
||||
const LEARNING_RATE: f64 = 0.001;
|
||||
const EPOCHS: usize = 100000;
|
||||
|
||||
for epoch in 0..EPOCHS {
|
||||
let predictions = model.forward(inputs.clone());
|
||||
|
||||
Reference in New Issue
Block a user