Skip to content

About the GPU resource #19

Description

@jiangshibiao

Hello! I'm now trying to run this code on CoSQL datasets but meet some problems.
I use the default parameters and my 16G GPU get out of memory. I have tried some adjustment but it didn't work. For example, batch_size is set to 16 in the default, and I still get out of memory even I change it to 1.(After I read codes briefly, I found that this parameter will be overwritten by interaction_level. It is always 1 so it didn't work.)
Could you please share something you know, like how to reduce the GPU needs, or point out that I have made some mistakes.

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