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

Update batch sizes for standard torch networks #749

Merged
merged 1 commit into from
May 17, 2016

Conversation

lukeyeager
Copy link
Member

Make them match Caffe's from #733.

I tested them all on a GTX 980 (4GB card) and none ran out of memory.

@lukeyeager
Copy link
Member Author

@gheinrich this is ready for review

@gheinrich
Copy link
Contributor

That looks good to me thanks!

@lukeyeager lukeyeager merged commit 1aafc36 into NVIDIA:master May 17, 2016
@lukeyeager lukeyeager deleted the torch-updated-batch-sizes branch May 17, 2016 17:10
SlipknotTN pushed a commit to cynnyx/DIGITS that referenced this pull request Mar 30, 2017
Update batch sizes for standard torch networks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants