improved tests

2026-01-06 09:44:45 +01:00
parent ec5d259ff7
commit 42520e5761
3 changed files with 17 additions and 19 deletions
@@ -16,7 +16,7 @@ class TestModel(Model):
 		else:
 			# Hidden binary
 			self.unit1 = Entity((96 * 96, 333), EntityParams(do_gaussian_visible=True, do_gaussian_hidden=False),
-								TrainingParams(learning_rate=0.0001, momentum=0.9, num_epochs=1000))
+								TrainingParams(learning_rate=0.001, momentum=0.9, num_epochs=1000))

 	def forward(self, x: Mat):
 		x = self.unit1.forward(x)