Skip to content

在40G显存的机器上训练显存会爆 OOM #2823

@no-swing

Description

@no-swing

我用batch_type='token',batch_size=128这套超参去跑,直接OOM,数据集里最长音频大概20s,把batch_size再调小也会

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions