Error with compute_bn_statistics

alexgkendall / caffe-segnet

Implementation of SegNet: A Deep Convolutional Encoder-Decoder Architecture for Semantic Pixel-Wise Labelling

http://mi.eng.cam.ac.uk/projects/segnet/

Other

1.08k stars 452 forks source link

Error with compute_bn_statistics #116

Closed victoriajyang closed 7 years ago

victoriajyang commented 7 years ago

I'm experiencing some errors with computing bn_statistics, and I saw that previous fixes indicated to remove the accuracy and loss layers. However, doesn't the compute_bn_statistics script use these layers at line 144?

victoriajyang commented 7 years ago

If I only remove the accuracy layer, everything works as it should, except statistics are only computed for half of the images

victoriajyang commented 7 years ago

the issue was solved by decreasing the batch size to 1