Elly_KOP

Here is my code DeviceOption option; option.set_device_type(CUDA); new CUDAContext(option); // Load Squeezenet model NetDef init_net, predict_net; init_net.mutable_device_option()->set_device_type(CUDA); predict_net.mutable_device_option()->set_device_type(CUDA); // >>> with open(path_to_INIT_NET) as f: CAFFE_ENFORCE(ReadProtoFromFile(FLAGS_init_net, &init_net)); // >>> with open(path_to_PREDICT_NET)...