| | |
| | | network net = parse_network_cfg(cfgfile); |
| | | network sum = parse_network_cfg(cfgfile); |
| | | |
| | | char *weightfile = argv[4]; |
| | | char *weightfile = argv[4]; |
| | | load_weights(&sum, weightfile); |
| | | |
| | | int i, j; |
| | | int n = argc - 5; |
| | | for(i = 0; i < n; ++i){ |
| | | weightfile = argv[i+5]; |
| | | weightfile = argv[i+5]; |
| | | load_weights(&net, weightfile); |
| | | for(j = 0; j < net.n; ++j){ |
| | | layer l = net.layers[j]; |
| | |
| | | |
| | | int main(int argc, char **argv) |
| | | { |
| | | #ifdef _DEBUG |
| | | _CrtSetDbgFlag(_CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF); |
| | | #endif |
| | | |
| | | int i; |
| | | for (i = 0; i < argc; ++i) { |
| | | if (!argv[i]) continue; |
| | | strip_args(argv[i]); |
| | | } |
| | | |
| | | //test_resize("data/bad.jpg"); |
| | | //test_box(); |
| | | //test_convolutional_layer(); |
| | |
| | | #else |
| | | if(gpu_index >= 0){ |
| | | cuda_set_device(gpu_index); |
| | | check_error(cudaSetDeviceFlags(cudaDeviceScheduleBlockingSync)); |
| | | } |
| | | #endif |
| | | |