显存占用问题

[Ace-blue]

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

[gaoypeng]

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

[judyxt]

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