captcha_detection icon indicating copy to clipboard operation
captcha_detection copied to clipboard

大佬,请教下执行目标检测时如何开启使用GPU 共享内存呢?

Open LmingXie opened this issue 1 year ago • 1 comments

我在跑目标检测时发现GPU 共享内存使用率几乎没有,CPU 使用率也不高,可以通过参数启用共享内存吗? image

这是我的参数配置:


# 创建一个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

LmingXie avatar May 20 '23 05:05 LmingXie

共享内存正常情况下本来就不会主动调用的吧 image

1136623363 avatar Sep 05 '23 17:09 1136623363