Skip to content

About batch_size during training #58

@zhangziwenHIT

Description

@zhangziwenHIT

You mentioned in the paper that setting batch_size==8 on four 2080Ti takes about three days to train.
image
But when I set batch_size>=4 on V100, I get an error.
image
The problem is the same as in this issue.#23
image
Is there something wrong with my configuration?As far as I know, the 2080Ti has 11GB of memory, while the V100 I used has 32GB. This is puzzling to me.Looking forward to getting your answer, thank you!

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