xiansen11 comments

Repositories
Issues
Comments

Results 2 comments of


                                            xiansen11

GPU and Memory

> > Hi, have you made progress on running on a single 4090 GPU? Any help or guide would be appreciated! > > hello，我前段时间在单个3090上跑了一下，用的small版本的dataset。 Stage1: Perception training 跑的时候out of memory...

RuntimeError: The expanded size of the tensor (3) must match the existing size (4) at non-singleton dimension 3. Target sizes: [1, 1, 3, 3]. Tensor sizes: [4, 4]

> bda_[:, :, :, :] = bda.unsqueeze(1) 解决了，感谢大佬