YOLO-V3-IOU
YOLO-V3-IOU copied to clipboard
GPU
这个代码怎么限制GPU使用情况,我用 config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.1 session = tf.Session(config=config)
self.sess = K.get_session(session) 这样是错误的,作者大神有什么方法限制显存么