From 73f7aacf35ec9b1d0f9de9ddf38af0889f213e99 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 20 Sep 2016 18:34:49 +0000
Subject: [PATCH] better multigpu
---
src/dice.c | 5 ++---
1 files changed, 2 insertions(+), 3 deletions(-)
diff --git a/src/dice.c b/src/dice.c
index 6f148b0..2286459 100644
--- a/src/dice.c
+++ b/src/dice.c
@@ -6,7 +6,6 @@
void train_dice(char *cfgfile, char *weightfile)
{
- data_seed = time(0);
srand(time(0));
float avg_loss = -1;
char *base = basecfg(cfgfile);
@@ -27,7 +26,7 @@
while(1){
++i;
time=clock();
- data train = load_data(paths, imgs, plist->size, labels, 6, net.w, net.h);
+ data train = load_data_old(paths, imgs, plist->size, labels, 6, net.w, net.h);
printf("Loaded: %lf seconds\n", sec(clock()-time));
time=clock();
@@ -60,7 +59,7 @@
int m = plist->size;
free_list(plist);
- data val = load_data(paths, m, 0, labels, 6, net.w, net.h);
+ data val = load_data_old(paths, m, 0, labels, 6, net.w, net.h);
float *acc = network_accuracies(net, val, 2);
printf("Validation Accuracy: %f, %d images\n", acc[0], m);
free_data(val);
--
Gitblit v1.10.0