sequence_roi_val.py 文件源码

def load_nets(args, cur_gpu):
    # initialize solver and feature net,
    # RNN should be initialized before CNN, because CNN cudnn conv layers
    # may assume using all available memory
    caffe.set_mode_gpu()
    caffe.set_device(cur_gpu)
    net = caffe.Net(args.model, args.weights, caffe.TEST)
    return net