Commit c24a0047 authored by Carlos Alfredo Yeverino Rodriguez's avatar Carlos Alfredo Yeverino Rodriguez
Browse files

Minor change to test deploy net

parent de0f235f
Pipeline #75931 passed with stages
in 4 minutes and 50 seconds
......@@ -79,7 +79,6 @@ if __name__ == "__main__":
context = 'gpu'
<#--Code section that decides the mode cannot be moved into the load_net function since workspace.FeedBlob also needs this parameter-->
if context == 'cpu':
device_opts = core.DeviceOption(caffe2_pb2.CPU, 0)
......@@ -87,6 +86,7 @@ if __name__ == "__main__":
elif context == 'gpu':
device_opts = core.DeviceOption(caffe2_pb2.CUDA, 0)
print("GPU mode selected")
LeNet.load_net(LeNet.INIT_NET, LeNet.PREDICT_NET, device_opts=device_opts)
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment