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

请教一下大家，4层tinybert和6层tinybert加载到显存中，占用了多大的显存 #241

Open

louiss007 opened this issue Nov 21, 2023 · 0 comments

louiss007 commented Nov 21, 2023 •

edited

我用公开的6层tinybert，利用分类数据微调了一下，之后将模型加载到显存中，看到显存占用有1033M，大家的显存占用有多少呢，感觉占用还是有点多，模型才228M，为什么显存却占了1033M呢【非训练阶段，推理阶段】

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment