From 23955b9fa0a29465ad2a2d13c445b49e6d5adef2 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 08 Feb 2016 19:50:45 +0000
Subject: [PATCH] binary reading weights
---
src/rnn.c | 4 +---
1 files changed, 1 insertions(+), 3 deletions(-)
diff --git a/src/rnn.c b/src/rnn.c
index 3984d93..3865209 100644
--- a/src/rnn.c
+++ b/src/rnn.c
@@ -48,8 +48,6 @@
void train_char_rnn(char *cfgfile, char *weightfile, char *filename)
{
FILE *fp = fopen(filename, "rb");
- //FILE *fp = fopen("data/ab.txt", "r");
- //FILE *fp = fopen("data/grrm/asoiaf.txt", "r");
fseek(fp, 0, SEEK_END);
size_t size = ftell(fp);
@@ -192,5 +190,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