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

显存占用问题 #18

Open

Ace-blue opened this issue Aug 26, 2024 · 2 comments

Ace-blue commented Aug 26, 2024

怎么这个显存的占用这么奇怪🤔

Owner

Hlings commented Aug 27, 2024

确实有点奇怪（我猜测是做feature extraction之后没有release 显存（因为代码里面会提前把feature算出来用于选择数据的那部分）

judyxt commented Sep 19, 2024

我在训练的时候显存反而不高（sim10k->city,bs=16），但是在一开始的数据加载中显存爆满，不知道是什么原因，有办法优化吗？

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