search

jzbontar / mc-cnn

Stereo Matching by Training a Convolutional Neural Network to Compare Image Patches
BSD 2-Clause "Simplified" License
707 stars 232 forks source link

min() function with vol:cuda() #50

Open Sarah20187 opened 6 years ago

Sarah20187 commented 6 years ago

after updating cudnn (v4 to v6), I found that the result of min() changed a lot in x range (parameter: -sm_terminate cnn).

Sarah20187 commented 6 years ago

I found the problem is caused by this

vol = torch.CudaTensor(1, disp_max, output:size(3), output:size(4)):fill(0/0);

different cudnn version set different values wiht '0/0'. So I just simply changed it to a large value and the problem solved. @jzbontar