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

求教 #1

Open
meox3259 opened this issue Apr 22, 2023 · 5 comments
Open

求教 #1

meox3259 opened this issue Apr 22, 2023 · 5 comments

Comments

@meox3259
Copy link

想问一下这个真的跑得起来吗，为啥我跑了下中间显存显示要2000个G
@h-z-h-cell
Copy link
Owner

我在我本地电脑上可以跑cifa10和cifar10dvs那两个，imagenet我没试过。

@meox3259
Copy link
Author

我用imagenet能跑通，但是cifar不太行...

@meox3259
Copy link
Author

想问下是直接用模型跑就行吗，不需要修改

@h-z-h-cell
Copy link
Owner

我电脑上就是直接用这个代码这么跑的，只不过数据集我没有放上来，需要自己下载。我设置batch_size为128的时候会提示我内存60G不够用，我换成batch_size为64才能跑，如果跑不动，试试要不要再缩小一点batch_size。

@meox3259
Copy link
Author

感谢

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
2 participants
@meox3259 @h-z-h-cell