Skip to content

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

@LmingXie

Description

@LmingXie

我在跑目标检测时发现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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions