Procházet zdrojové kódy

perplexity : fix winogrande N tasks option

Georgi Gerganov před 2 roky
rodič
revize
d391ae9b49
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1 1
      examples/perplexity/perplexity.cpp

+ 1 - 1
examples/perplexity/perplexity.cpp

@@ -865,7 +865,7 @@ static void winogrande_score(llama_context * ctx, const gpt_params & params) {
         }
         }
         float scale = 1/(1.f + (float)rng.max());
         float scale = 1/(1.f + (float)rng.max());
         std::vector<winogrande_entry> selected;
         std::vector<winogrande_entry> selected;
-        selected.reserve(params.winogrande_tasks);
+        selected.resize(params.winogrande_tasks);
         for (int i = 0; i < int(params.winogrande_tasks); ++i) {
         for (int i = 0; i < int(params.winogrande_tasks); ++i) {
             int j = int(scale*rng()*aux.size());
             int j = int(scale*rng()*aux.size());
             selected[i] = std::move(data[aux[j]]);
             selected[i] = std::move(data[aux[j]]);