small bug in forward_network_gpu
| | |
| | | //net.seen=0; |
| | | int imgs = 1024; |
| | | int i = net.seen/imgs; |
| | | int solved = 0; |
| | | int solved = 1; |
| | | list *plist; |
| | | char **labels = get_labels("/data/captcha/reimgs.labels2.list"); |
| | | char **labels = get_labels("/data/captcha/reimgs.labels.list"); |
| | | if (solved){ |
| | | plist = get_paths("/data/captcha/reimgs.solved.list"); |
| | | }else{ |
| | |
| | | set_batch_network(&net, 1); |
| | | srand(2222222); |
| | | int i = 0; |
| | | char **names = get_labels("/data/captcha/reimgs.labels2.list"); |
| | | char **names = get_labels("/data/captcha/reimgs.labels.list"); |
| | | char input[256]; |
| | | int indexes[26]; |
| | | while(1){ |
| | |
| | | avg_loss = avg_loss*.9 + loss*.1; |
| | | printf("%d: %f, %f avg, %lf seconds, %d images\n", i, loss, avg_loss, sec(clock()-time), net.seen); |
| | | free_data(train); |
| | | if((i % 30000) == 0) net.learning_rate *= .1; |
| | | if((i % 35000) == 0) net.learning_rate *= .1; |
| | | if(i%1000==0){ |
| | | char buff[256]; |
| | | sprintf(buff, "/home/pjreddie/imagenet_backup/%s_%d.weights",base, i); |
| | |
| | | int i; |
| | | for(i = 0; i < net.n; ++i){ |
| | | layer l = net.layers[i]; |
| | | if(l.delta){ |
| | | if(l.delta_gpu){ |
| | | scal_ongpu(l.outputs * l.batch, 0, l.delta_gpu, 1); |
| | | } |
| | | if(l.type == CONVOLUTIONAL){ |