Open jiaxinxie97 opened 4 years ago
Hi, chao I tried to train neural_rgbd on eigen split, but can't get reasonable results. So I am wondering what's your batch_size setting for kitti dataset. This line https://github.com/NVlabs/neuralrgbd/blob/c8071a0bcbd4c4e7ef95c44e7de9c51353ab9764/code/mdataloader/batch_loader.py#L121 enforce me to set a batch_size>1. And this line https://github.com/NVlabs/neuralrgbd/blob/c8071a0bcbd4c4e7ef95c44e7de9c51353ab9764/code/models/basic.py#L240 enforce me to set batch_size=1. So I comment the former one and set batch_size=1, that's the only change I made to the code, kitti raw data was directly downloaded. Is this different from your training?
Hi, chao I tried to train neural_rgbd on eigen split, but can't get reasonable results. So I am wondering what's your batch_size setting for kitti dataset. This line https://github.com/NVlabs/neuralrgbd/blob/c8071a0bcbd4c4e7ef95c44e7de9c51353ab9764/code/mdataloader/batch_loader.py#L121 enforce me to set a batch_size>1. And this line https://github.com/NVlabs/neuralrgbd/blob/c8071a0bcbd4c4e7ef95c44e7de9c51353ab9764/code/models/basic.py#L240 enforce me to set batch_size=1. So I comment the former one and set batch_size=1, that's the only change I made to the code, kitti raw data was directly downloaded. Is this different from your training?