Skip to content

GPU 显存问题 #26

@ume-technology

Description

@ume-technology

我想走一下 Bert + CNN 的SST-2 数据的分类效果。
但是上来就报出 GPU 溢出,我换了公司服务器还是溢出,公司显存 10.75 G
我想知道这是为什么,是确实溢出还是说代码哪里我需要调整呢?

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