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/cnn.c | 8 ++++----
1 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/src/cnn.c b/src/cnn.c
index 7971b95..620126d 100644
--- a/src/cnn.c
+++ b/src/cnn.c
@@ -374,7 +374,7 @@
void train_imagenet_distributed(char *address)
{
float avg_loss = 1;
- srand(0);
+ srand(time(0));
network net = parse_network_cfg("cfg/alexnet.client");
printf("Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay);
int imgs = 1000/net.batch+1;
@@ -412,11 +412,11 @@
{
float avg_loss = 1;
//network net = parse_network_cfg("/home/pjreddie/imagenet_backup/alexnet_1270.cfg");
- srand(0);
+ srand(time(0));
network net = parse_network_cfg("cfg/alexnet.cfg");
printf("Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay);
int imgs = 1000/net.batch+1;
- imgs=1;
+ //imgs=1;
int i = 0;
char **labels = get_labels("/home/pjreddie/data/imagenet/cls.labels.list");
list *plist = get_paths("/data/imagenet/cls.train.list");
@@ -872,7 +872,7 @@
void run_server()
{
- srand(0);
+ srand(time(0));
network net = parse_network_cfg("cfg/alexnet.server");
server_update(net);
}
--
Gitblit v1.10.0