search

kakaxi314 / BP-Net

Implementation of our paper 'Bilateral Propagation Network for Depth Completion'
MIT License
84 stars 7 forks source link

About Computer Crash Problems #21

Open yiandjer opened 2 months ago

yiandjer commented 2 months ago

Hi author, thanks for the code, I have successfully reproduced it using the kitti dataset, but I batch_size=4,at first it works fine, after a while the computer gets stuck and crashes, what's the reason for that, is it that the batch_size is too big (my computer is using an nvidia 3090)

yiandjer commented 2 months ago

我把batch_size改成了2,还是死机了（一块3090训练）