1 parent 9b7212b commit ac57012Copy full SHA for ac57012
1 file changed
LycorisNet/sources/lycoris.cpp
@@ -715,13 +715,15 @@ namespace LycorisNet {
715
// The functor of computeBatch().
716
void Lycoris::computeBatchCore(uint32_t start, uint32_t end, float **output) {
717
auto temp = best->clone();
718
+
719
for (uint32_t i = start; i < end; ++i) {
720
temp->forward((*data_for_computeBatch)[i], output[i]);
721
722
if (args->mode == "classify") {
723
LycorisUtils::softmax(output[i], outputNum);
724
}
725
726
727
delete temp;
728
729
0 commit comments