Commit 5d03b99a authored by Davis King's avatar Davis King

Changed to avoid compiler warning.

parent 02ed083c
...@@ -279,7 +279,7 @@ int main(int argc, char** argv) try ...@@ -279,7 +279,7 @@ int main(int argc, char** argv) try
} }
// a mini-batch smaller than the default can be used with GPUs having less memory // a mini-batch smaller than the default can be used with GPUs having less memory
const int minibatch_size = argc == 3 ? std::stoi(argv[2]) : 23; const unsigned int minibatch_size = argc == 3 ? std::stoi(argv[2]) : 23;
cout << "mini-batch size: " << minibatch_size << endl; cout << "mini-batch size: " << minibatch_size << endl;
const double initial_learning_rate = 0.1; const double initial_learning_rate = 0.1;
......
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