Can I use 2 different networks(CPU/GPU) within one python script?

16 views
Skip to first unread message

Alex Orloff

unread,
Nov 7, 2017, 5:39:31 PM11/7/17
to Caffe Users
Hi,
I need to process big images and small ones as well
So I want to run 2 networks - CPU based for small and GPU based for big
How can I do it with pycaffe?
I see caffe.set_mode_cpu() and caffe.set_mode_gpu()
So do I need to 
import caffe
twice?

Alex Orloff

unread,
Nov 7, 2017, 7:40:44 PM11/7/17
to Caffe Users
I tried
import caffe as GPUCaffe
import caffe as CPUCaffe
CPUcaffe.set_mode_cpu()
GPUcaffe.set_mode_gpu()

but it doesn't work
Reply all
Reply to author
Forward
0 new messages