def init_testnet(test_net, trained_model=None, test_device=0):
caffe.set_mode_gpu()
caffe.select_device(test_device, False)
if trained_model is None:
return caffe.Net(test_net, caffe.TEST)
else:
return caffe.Net(test_net, trained_model, caffe.TEST)
def init_solver(solver_config, options):
caffe.set_mode_gpu()
caffe.select_device(options.train_device, False)
solver_inst = caffe.get_solver(solver_config)
if options.test_net is None:
return solver_inst, None
else:
return solver_inst, init_testnet(options.test_net, test_device=options.test_device)