Skip to content

GPU utilization #16

Description

@hannahmenke

Hi,

First, thanks for developing such a great model.

I am testing training the model based on the test NMC image on a quadro p5000 before I move on to my own images. With the default settings the model uses about 5GB of GPU RAM and only 25% of GPU compute capacity. I have increased both the batch_size and D_batch_size to 24 since my GPU has 16GB of RAM. However, this has not made the model use any more GPU compute capacity (only increased the RAM usage). Is there something else I should change to allow the GPU to utilize it's full power?

Thanks,

Hannah

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

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