From d81ab1942be926573a93649a558ee9b2d1156a33 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 08 Dec 2014 07:59:45 +0000
Subject: [PATCH] mnist now
---
src/server.c | 4 +++-
1 files changed, 3 insertions(+), 1 deletions(-)
diff --git a/src/server.c b/src/server.c
index c802f84..657ea7c 100644
--- a/src/server.c
+++ b/src/server.c
@@ -9,6 +9,7 @@
#include "mini_blas.h"
#include "utils.h"
+#include "parser.h"
#include "server.h"
#include "connected_layer.h"
#include "convolutional_layer.h"
@@ -82,7 +83,6 @@
connection_info info = *(connection_info *) pointer;
int fd = info.fd;
network net = info.net;
- ++*(info.counter);
int i;
for(i = 0; i < net.n; ++i){
if(net.types[i] == CONVOLUTIONAL){
@@ -117,6 +117,8 @@
}
printf("Received updates\n");
close(fd);
+ ++*(info.counter);
+ if(*(info.counter)%10==0) save_network(net, "/home/pjreddie/imagenet_backup/alexnet.part");
}
void server_update(network net)
--
Gitblit v1.10.0