captcha_detection
captcha_detection copied to clipboard
大佬,请教下执行目标检测时如何开启使用GPU 共享内存呢?
我在跑目标检测时发现GPU 共享内存使用率几乎没有,CPU 使用率也不高,可以通过参数启用共享内存吗?
这是我的参数配置:
# 创建一个DefaultYolo3Config的子类,在子类里覆盖默认的配置
class MyXYoloConfig(DefaultYolo3Config):
def __init__(self):
super(MyXYoloConfig, self).__init__()
# 数据集路径,推荐使用绝对路径
self._dataset_path = './xyolo_label.txt'
# 类别名称文件路径,推荐使用绝对路径
self._classes_path = './classes.txt'
# 模型保存路径,默认是保存在当前路径下的xyolo_data下的,也可以进行更改
# 推荐使用绝对路径
self._output_model_path = 'detect_model.h5'
# 解冻时,训练的batch_size.注意,解冻时训练对GPU内存需求量非常大,这里建议设置小一点
self.unfreeze_batch_size = 2
共享内存正常情况下本来就不会主动调用的吧