layer changes

Benny-Nottonson · Feb 4, 2024 · 567cb41 · 567cb41
1 parent 8f184a0
commit 567cb41
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test_layer.mojo b/test_layer.mojo
@@ -46,7 +46,7 @@ fn main() raises:
 
     var avg_loss: Float32 = 0.0
     let every = 1000
-    let num_epochs = 20000
+    let num_epochs = 200000
 
     var input = Tensor[data_shape]().initialize[RandomUniform, 0, 1]()
     let true_vals = Tensor[data_shape]().initialize[RandomUniform, 0, 1]()