| | |
| | | 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); |
| | |
| | | fprintf(stderr, "Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay); |
| | | int batch = net.batch; |
| | | int steps = net.time_steps; |
| | | //*net.seen = 0; |
| | | int i = (*net.seen)/net.batch; |
| | | |
| | | clock_t time; |
| | |
| | | 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); |
| | | } |