AlexeyAB
2018-02-21 e4ab47dfcedb4c87e5eddf484caa4ac0c020fc9b
src/network.c
@@ -378,6 +378,7 @@
    }
#ifdef GPU
    if(gpu_index >= 0){
      printf(" try to allocate workspace, ");
        net->workspace = cuda_make_array(0, (workspace_size-1)/sizeof(float)+1);
      printf(" CUDA allocate done! \n");
    }else {