请问可以将batchsize设置为大于1吗？

[pingyang97]

如何设置呢？我将config文件中的samples_per_gpu参数修改后还是会报错

[weiyithu]

当时这套代码是在3090上跑的，只考虑了每张卡batchsize为1的情况，很抱歉暂不支持多batchsize

[pingyang97]

请问主要是哪些因素限制了 bs的设置呢？

…

---- 回复的原邮件 ---- | 发件人 | Yi ***@***.***> | | 日期 | 2023年11月20日 10:36 | | 收件人 | ***@***.***> | | 抄送至 | ***@***.***>***@***.***> | | 主题 | Re: [weiyithu/SurroundOcc] 请问可以将batchsize设置为大于1吗？ (Issue #79) | 当时这套代码是在3090上跑的，只考虑了每张卡batchsize为1的情况，很抱歉暂不支持多batchsize — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***>

[weiyithu]

同学你改了samples_per_gpu后报什么错

[HuaiyuanXu]

同学你改了samples_per_gpu后报什么错

RuntimeError: stack expects each tensor to be equal size, but got [40086, 4] at entry 0 and [42776, 4] at entry 1

改了samples_per_gpu=2之后，两个张量在堆叠（stack）操作时的维度不一致。这里的张量指的gt_occ。

[pingyang97]

gt_occ的尺寸不一致，放入同一个batch会报错

[weiyithu]

我猜测应该是dataloader那块出了问题，可以试试把这里 改成list的形式