Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

How much GPU memory is required #5

Closed
amiltonwong opened this issue Nov 11, 2017 · 1 comment
Closed

How much GPU memory is required #5

amiltonwong opened this issue Nov 11, 2017 · 1 comment

Comments

@amiltonwong
Copy link

Hi, authors,

How much GPU memory is required, in training and prediction, respectively?
Does a 6GB GPU card meet the requirement?

THX!
@GrassSunFlower
Copy link
Contributor

It depends on your patch size and way of usage. In terms of training, a 320*320 patch with batch size 1 should take about 3GB. As for testing, a full resolution of 2048 * 1024 should take above 6GB GPU memory. So if you want to get better results, I would highly recommend you use a GPU with more than 8GB memory.

@wpqmanu wpqmanu closed this as completed Nov 14, 2017
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@wpqmanu @amiltonwong @GrassSunFlower