| | |
| | | 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); |
| | |
| | | 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) |