From 76ee68f96d864a27312c9aa09856ddda559a5cd9 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 28 Aug 2014 02:11:46 +0000
Subject: [PATCH] Trying some stuff w/ dropout
---
src/data.c | 16 +++++++++++++---
1 files changed, 13 insertions(+), 3 deletions(-)
diff --git a/src/data.c b/src/data.c
index 846b950..aa8fecf 100644
--- a/src/data.c
+++ b/src/data.c
@@ -148,6 +148,16 @@
return d;
}
+void get_batch(data d, int n, float *X, float *y)
+{
+ int j;
+ for(j = 0; j < n; ++j){
+ int index = rand()%d.X.rows;
+ memcpy(X+j*d.X.cols, d.X.vals[index], d.X.cols*sizeof(float));
+ memcpy(y+j*d.y.cols, d.y.vals[index], d.y.cols*sizeof(float));
+ }
+}
+
data load_all_cifar10()
{
data d;
@@ -158,7 +168,7 @@
d.X = X;
d.y = y;
-
+
for(b = 0; b < 5; ++b){
char buff[256];
sprintf(buff, "data/cifar10/data_batch_%d.bin", b+1);
@@ -176,8 +186,8 @@
fclose(fp);
}
//normalize_data_rows(d);
- translate_data_rows(d, -144);
- scale_data_rows(d, 1./128);
+ translate_data_rows(d, -144);
+ scale_data_rows(d, 1./128);
return d;
}
--
Gitblit v1.10.0