Commit 14acae38 authored by Davis King's avatar Davis King

Made unit tests more reliable

parent 31280b54
...@@ -2143,7 +2143,7 @@ namespace ...@@ -2143,7 +2143,7 @@ namespace
trainer.set_learning_rate(1e-5); trainer.set_learning_rate(1e-5);
trainer.set_min_learning_rate(1e-11); trainer.set_min_learning_rate(1e-11);
trainer.set_mini_batch_size(50); trainer.set_mini_batch_size(50);
trainer.set_max_num_epochs(300); trainer.set_max_num_epochs(2000);
trainer.train(x, y); trainer.train(x, y);
running_stats<double> rs; running_stats<double> rs;
...@@ -2228,6 +2228,8 @@ namespace ...@@ -2228,6 +2228,8 @@ namespace
{ {
print_spinner(); print_spinner();
srand(1234);
const int output_width = 7; const int output_width = 7;
const int output_height = 7; const int output_height = 7;
const int num_samples = 100; const int num_samples = 100;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment