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

gen_rnet_data #9

Open
Sarah-Elfanty opened this issue Apr 19, 2019 · 2 comments
Open

gen_rnet_data #9

Sarah-Elfanty opened this issue Apr 19, 2019 · 2 comments

Comments

@Sarah-Elfanty
Copy link

Hi, I've been trying to train this model from scratch on a local dataset, but I've been encountering a CUDA memory issue when i run gen_Rnet_train_data.py
I'm using a 1050ti , 4gb GPU with 16 gb RAM.
@Sierkinhane
Copy link
Owner

Maybe you should reduce the batch size of testimageloader

@liaomuqin
Copy link

我也遇到了同样的问题，请问怎么解决。。。

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
@Sarah-Elfanty @Sierkinhane @liaomuqin