Ruslan_Lenin
2018-06-29 eae7b34cc0906268ddef2e64551ea5ca0951c0c8
src/network.c
@@ -443,8 +443,8 @@
    }
#ifdef GPU
    if(gpu_index >= 0){
      printf(" try to allocate workspace = %zu * sizeof(float), ", (workspace_size - 1) / sizeof(float) + 1);
        net->workspace = cuda_make_array(0, (workspace_size-1)/sizeof(float)+1);
      printf(" try to allocate workspace = %zu * sizeof(float), ", workspace_size / sizeof(float) + 1);
        net->workspace = cuda_make_array(0, workspace_size/sizeof(float) + 1);
      printf(" CUDA allocate done! \n");
    }else {
        free(net->workspace);
@@ -493,6 +493,11 @@
    return def;
}
layer* get_network_layer(network* net, int i)
{
    return net->layers + i;
}
image get_network_image(network net)
{
    int i;