From 01f20abe847768a8fb3e27739a284b541eb08ff6 Mon Sep 17 00:00:00 2001 From: Joseph Redmon Date: Fri, 5 Feb 2016 12:56:18 -0800 Subject: [PATCH] generate --- src/rnn.c | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/rnn.c b/src/rnn.c index 43cac5a7..3865209b 100644 --- a/src/rnn.c +++ b/src/rnn.c @@ -190,5 +190,5 @@ void run_char_rnn(int argc, char **argv) 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); }