mirror of
https://github.com/davisking/dlib.git
synced 2024-11-01 10:14:53 +08:00
Added a comment
This commit is contained in:
parent
963e8e8266
commit
64052a3ff1
@ -142,6 +142,8 @@ int main(int argc, char** argv) try
|
||||
|
||||
// Now we are ready to create our network and trainer.
|
||||
net_type net(options);
|
||||
// The MMOD loss requires that the number of filters in the final network layer equal
|
||||
// options.detector_windows.size(). So we set that here as well.
|
||||
net.subnet().layer_details().set_num_filters(options.detector_windows.size());
|
||||
dnn_trainer<net_type> trainer(net);
|
||||
trainer.set_learning_rate(0.1);
|
||||
|
Loading…
Reference in New Issue
Block a user