OrgLion-ML
/
dlib
mirror of https://github.com/davisking/dlib.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Tweak default minibatch size

This commit is contained in:
Juha Reunanen 2019-10-28 06:48:12 +02:00
parent 414b5543f6
commit edb083709a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/dnn_instance_segmentation_train_ex.cpp
View File

@ -550,7 +550,7 @@ int main(int argc, char** argv) try
}
// mini-batches smaller than the default can be used with GPUs having less memory
const unsigned int det_minibatch_size = argc >= 3 ? std::stoi(argv[2]) : 75;
const unsigned int det_minibatch_size = argc >= 3 ? std::stoi(argv[2]) : 40;
const unsigned int seg_minibatch_size = argc >= 4 ? std::stoi(argv[3]) : 25;
cout << "det mini-batch size: " << det_minibatch_size << endl;
cout << "seg mini-batch size: " << seg_minibatch_size << endl;