forked from StrafesNET/strafe-ai
change learning rate
This commit is contained in:
@@ -325,7 +325,7 @@ fn training() {
|
||||
&device,
|
||||
);
|
||||
|
||||
const LEARNING_RATE: f64 = 0.5;
|
||||
const LEARNING_RATE: f64 = 0.01;
|
||||
const EPOCHS: usize = 10000;
|
||||
|
||||
for epoch in 0..EPOCHS {
|
||||
|
||||
Reference in New Issue
Block a user