From 9dd0cab9dc73657e19e2a9cc372dd27311103c5e Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 21 Jul 2015 23:13:41 +0000
Subject: [PATCH] Multiple GPUs instead of just sm_52

---
 src/matrix.h |    2 +-
 1 files changed, 1 insertions(+), 1 deletions(-)

diff --git a/src/matrix.h b/src/matrix.h
index 01d825d..d84431c 100644
--- a/src/matrix.h
+++ b/src/matrix.h
@@ -11,7 +11,7 @@
 
 matrix csv_to_matrix(char *filename);
 matrix hold_out_matrix(matrix *m, int n);
-float matrix_accuracy(matrix truth, matrix guess);
+float matrix_topk_accuracy(matrix truth, matrix guess, int k);
 void matrix_add_matrix(matrix from, matrix to);
 
 float *pop_column(matrix *m, int c);

--
Gitblit v1.10.0