Joseph Redmon
2016-02-05 baec87e7ac924ba5c4d8489fa3163d84dbadfe75
more rnn stuff
2 files modified
4 ■■■■ changed files
src/rnn.c 2 ●●●●● patch | view | raw | blame | history
src/rnn_layer.c 2 ●●●●● patch | view | raw | blame | history
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);
src/rnn_layer.c
@@ -18,10 +18,12 @@
    l->x += num;
    l->x_norm += num;
#ifdef GPU
    l->output_gpu += num;
    l->delta_gpu += num;
    l->x_gpu += num;
    l->x_norm_gpu += num;
#endif
}
layer make_rnn_layer(int batch, int inputs, int hidden, int outputs, int steps, ACTIVATION activation, int batch_normalize, int log)