From a99050f0c8cb0315fa31e3d1fa3e38594fe5e40a Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 08 Dec 2014 04:16:21 +0000
Subject: [PATCH] Some fixes to momentum
---
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