Joseph Redmon
2014-10-26 edbccdfcaf46f11e631afe98796f3e6e170da5d0
src/cnn.c
@@ -499,7 +499,7 @@
    int iters = 10000/net.batch;
    while(++count <= 2000){
        clock_t start = clock(), end;
        float loss = train_network_sgd(net, train, iters);
        float loss = train_network_sgd_gpu(net, train, iters);
        end = clock();
        float test_acc = network_accuracy(net, test);
        //float test_acc = 0;
@@ -957,8 +957,9 @@
int main(int argc, char *argv[])
{
    //test_gpu_blas();
    train_imagenet();
    test_gpu_blas();
    //train_imagenet();
    //train_nist();
    fprintf(stderr, "Success!\n");
    return 0;
}