From a392bbd0c957a00e3782c96e7ced84a29ff9dd88 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 15 Mar 2016 05:33:02 +0000
Subject: [PATCH] Play along w/ alphago
---
src/rnn.c | 3 ++-
1 files changed, 2 insertions(+), 1 deletions(-)
diff --git a/src/rnn.c b/src/rnn.c
index 43cac5a..30fa4bd 100644
--- a/src/rnn.c
+++ b/src/rnn.c
@@ -71,6 +71,7 @@
fprintf(stderr, "Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay);
int batch = net.batch;
int steps = net.time_steps;
+ //*net.seen = 0;
int i = (*net.seen)/net.batch;
clock_t time;
@@ -190,5 +191,5 @@
char *weights = (argc > 4) ? argv[4] : 0;
if(0==strcmp(argv[2], "train")) train_char_rnn(cfg, weights, filename);
else if(0==strcmp(argv[2], "valid")) valid_char_rnn(cfg, weights);
- else if(0==strcmp(argv[2], "test")) test_char_rnn(cfg, weights, len, seed, temp, rseed);
+ else if(0==strcmp(argv[2], "generate")) test_char_rnn(cfg, weights, len, seed, temp, rseed);
}
--
Gitblit v1.10.0