Commit d31723ff authored by Davis King's avatar Davis King

Fixed typo in example

parent b73f5e81
......@@ -202,12 +202,12 @@ int main(int argc, char** argv) try
// The dnn_trainer will use SGD by default, but you can tell it to use
// different solvers like adam.
//dnn_trainer<net_type,adam> trainer(net,adam(0.001));
dnn_trainer<net_type,adam> trainer(net,adam(0.001));
// Also, if you have multiple graphics cards you can tell the trainer to use
// them together to make the training faster. For example, replacing the
// above constructor call with this one would cause it to use GPU cards 0
// and 1.
dnn_trainer<net_type,adam> trainer(net,adam(0.001), {0,1});
//dnn_trainer<net_type,adam> trainer(net,adam(0.001), {0,1});
trainer.be_verbose();
trainer.set_synchronization_file("mnist_resnet_sync", std::chrono::seconds(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