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

求助compression_imagenet.py实验环境 #14

Open
lidanxu opened this issue Jan 13, 2020 · 2 comments
Open

求助compression_imagenet.py实验环境 #14

lidanxu opened this issue Jan 13, 2020 · 2 comments

Comments

@lidanxu
Copy link

lidanxu commented Jan 13, 2020

作者您好,我下载了您的代码以及 纯tensorflow实现的代码,发现这个模型很吃显存,我的环境是
image
但是只有把batch size设置为1才能跑,否则就超内存,可以请问你们实验环境是怎么样的,batch_size=1学不起来
但是我个人感觉这个模型没有那么复杂,层数也不多,不知道为何这么吃显存
@lidanxu
Copy link
Author

lidanxu commented Jan 13, 2020

希望您能给我一些帮助,我没搞明白为什么模型跑起来需要那么大的显存
这样我本地实验根本做不起来

@lidanxu
Copy link
Author

lidanxu commented Jan 13, 2020

万分感谢!

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
1 participant
@lidanxu