mirror of
https://github.com/davisking/dlib.git
synced 2024-11-01 10:14:53 +08:00
Fixed typo in example
This commit is contained in:
parent
b73f5e8160
commit
d31723ff45
@ -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));
|
||||
|
Loading…
Reference in New Issue
Block a user