Skip to content

Cannot run train_gan_net model #4

@hoangminhtoan

Description

@hoangminhtoan

Hi, I tried to reproduce the model but it ran out of CUDA memory. I run on quadro rtx 6000 (24GB). Even I change batch_size = 1 instead of 4. Can you provide your environment configurations?
Thank you!
image

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