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.

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

How to reduce the batch size? #90

Open
Bird-NZ opened this issue Dec 21, 2021 · 1 comment
Open

How to reduce the batch size? #90

Bird-NZ opened this issue Dec 21, 2021 · 1 comment

Comments

@Bird-NZ
Copy link

Bird-NZ commented Dec 21, 2021

I'm wanting to reduce the batch size in order to get around the dreaded "CUDA out of memory" error. I'm wanting to do this to be able to generate much larger images as outputs at least ~10meg. I don't see an Argument for this so thought perhaps this is hard coded?

@ProGamerGov
Copy link
Owner

ProGamerGov commented Dec 21, 2021

@Bird-NZ The batch size is always 1 for neural-style-pt, as batch size refers to the number of images being run through the network. You can reduce memory usage by using the Adam optimizer, and by using different models for later steps.

Maybe try the Channel Pruning model for the larger steps: https://github.com/ProGamerGov/neural-style-pt/wiki/Other-Models

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants