Join GitHub today
GitHub is home to over 28 million developers working together to host and review code, manage projects, and build software together.Sign up
Squeezenet Tutorial: Predictor outputs initialization information and exits #1352
Similar to #799, I am trying to run through the Loading Pre-Trained Models Tutorial with GPU/CUDA/Windows10: https://caffe2.ai/docs/tutorial-loading-pre-trained-models.html
The following code executes:
`device_opts = core.DeviceOption(caffe2_pb2.CUDA, 0)
init_def = caffe2_pb2.NetDef()
net_def = caffe2_pb2.NetDef()
print 'Running net...'
However, the system outputs a bunch of data about the model/predictor and then immediately exits without running the rest of the code or providing any additional information as to why the system has exited. Has anyone seen this behavior before?
Okay, I was able to capture the error:
Worked this out in CPU Model, scrapped the .pb I had and re-pulled from the github models repo (updated init to exec in your code); however, in GPU mode. I get an error when switching to GPU/CUDA
Does this suggest that Predictor only works on the CPU?