projects / platform / upstream / caffe.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 65c7fa6) | patch
Make classification.bin support models with less than 5 classes
authorFelix Abecassis <fabecassis@nvidia.com>
Fri, 14 Aug 2015 18:15:56 +0000 (11:15 -0700)
committerFelix Abecassis <fabecassis@nvidia.com>
Fri, 14 Aug 2015 18:22:38 +0000 (11:22 -0700)
commit98593e3aa64d9a8f42723fb03fa46a1343e12c33
tree69d00b39723773de8270d96adbfb64901616bfe8tree | snapshot
parent65c7fa6e92d81c68d44caa206bd003013786ac69commit | diff
Make classification.bin support models with less than 5 classes

The example program would crash if the number of classes was less than
5, since it was still attempting to get the top 5 predictions.

Close #2585
examples/cpp_classification/classification.cpp diff | blob | history
Domain: Machine Learning / ML Framework;