That's a so great work,But i have a question . When using mvdream for 3D reconstruction, an error will be reported when batchsize is set to 4. Can batchsize only be set to 8?

bytedance / MVDream

Multi-view Diffusion for 3D Generation

MIT License

715 stars 52 forks source link

That's a so great work,But i have a question . When using mvdream for 3D reconstruction, an error will be reported when batchsize is set to 4. Can batchsize only be set to 8? #15

Open xzr52 opened 9 months ago

xzr52 commented 9 months ago

seasonSH commented 9 months ago

it should support any multiples of 4. It looks like your input embedding batch size is wrong. The batch size of image is 8 but time/camera embedding somehow is 2.