From a6c51e3b758aee7fd3a6f1d37daa8dcad4891e52 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Thu, 08 Mar 2018 22:42:58 +0000
Subject: [PATCH] Fixes for small objects. And fixes of densenet201_yolo.cfg, resnet50_yolo.cfg, resnet152_yolo.cfg
---
src/rnn.c | 10 ++++------
1 files changed, 4 insertions(+), 6 deletions(-)
diff --git a/src/rnn.c b/src/rnn.c
index 5e229ba..eca6f55 100644
--- a/src/rnn.c
+++ b/src/rnn.c
@@ -129,7 +129,6 @@
void train_char_rnn(char *cfgfile, char *weightfile, char *filename, int clear, int tokenized)
{
srand(time(0));
- data_seed = time(0);
unsigned char *text = 0;
int *tokens = 0;
size_t size;
@@ -199,7 +198,7 @@
}
}
- if(i%100==0){
+ if(i%1000==0){
char buff[256];
sprintf(buff, "%s/%s_%d.weights", backup_directory, base, i);
save_weights(net, buff);
@@ -280,7 +279,7 @@
printf("\n");
}
-void test_tactic_rnn(char *cfgfile, char *weightfile, int num, char *seed, float temp, int rseed, char *token_file)
+void test_tactic_rnn(char *cfgfile, char *weightfile, int num, float temp, int rseed, char *token_file)
{
char **tokens = 0;
if(token_file){
@@ -301,9 +300,8 @@
int i, j;
for(i = 0; i < net.n; ++i) net.layers[i].temperature = temp;
int c = 0;
- int len = strlen(seed);
float *input = calloc(inputs, sizeof(float));
- float *out;
+ float *out = 0;
while((c = getc(stdin)) != EOF){
input[c] = 1;
@@ -490,5 +488,5 @@
else if(0==strcmp(argv[2], "validtactic")) valid_tactic_rnn(cfg, weights, seed);
else if(0==strcmp(argv[2], "vec")) vec_char_rnn(cfg, weights, seed);
else if(0==strcmp(argv[2], "generate")) test_char_rnn(cfg, weights, len, seed, temp, rseed, tokens);
- else if(0==strcmp(argv[2], "generatetactic")) test_tactic_rnn(cfg, weights, len, seed, temp, rseed, tokens);
+ else if(0==strcmp(argv[2], "generatetactic")) test_tactic_rnn(cfg, weights, len, temp, rseed, tokens);
}
--
Gitblit v1.10.0