google-research / sparf

This is the official code release for SPARF: Neural Radiance Fields from Sparse and Noisy Poses [CVPR 2023-Highlight]
https://prunetruong.com/sparf.github.io/
Apache License 2.0
285 stars 15 forks source link

Questions about batch setting and out of memory. #8

Closed yfqi closed 1 year ago

yfqi commented 1 year ago

I have a question: When I train my own dataset with joint_pose_nerf_training and train_sub = 6, it works fine. However when I train the network with nerf_training_w_gt_poses and train_sun=6, it shows out of memory. So:

  1. I wonder what differences about dataloaders between them two and what I can do to solve it.
  2. I also wonder how I can set the batches in your networks. Thank you
PruneTruong commented 1 year ago

Hi,

That is a bit strange, the dataloaders should be the same. I will check what could be different. As for the batches, you need to change config.rand_rays in the configs and reduce that number.

yfqi commented 1 year ago

您好,来信已收到,感谢您的邮件。

yfqi commented 1 year ago

您好,来信已收到,感谢您的邮件。