mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

accidentally broke multigpu a while ago :-/

Browse Source
This commit is contained in:
Joseph Redmon
2017-06-26 14:00:18 -07:00
parent 63ebd694d7
commit 59ed1719d4
12 changed files with 119 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/rnn.c
View File

@ -43,6 +43,7 @@ char **read_tokens(char *filename, size_t *read)
size = size*2;
d = realloc(d, size*sizeof(char *));
}
if(0==strcmp(line, "<NEWLINE>")) line = "\n";
d[count-1] = line;
}
fclose(fp);
@ -190,7 +191,7 @@ void train_char_rnn(char *cfgfile, char *weightfile, char *filename, int clear,
for(j = 0; j < streams; ++j){
//printf("%d\n", j);
if(rand()%10 == 0){
if(rand()%64 == 0){
//fprintf(stderr, "Reset\n");
offsets[j] = rand_size_t()%size;
reset_rnn_state(net, j);