TimoSaemann / caffe-segnet-cudnn5

This repository was a fork of BVLC/caffe and includes the upsample, bn, dense_image_data and softmax_with_loss (with class weighting) layers of caffe-segnet (https://github.com/alexgkendall/caffe-segnet) to run SegNet with cuDNN version 5.
Other
176 stars 128 forks source link

Error with compute_bn_statistics.py #25

Closed vsuryamurthy closed 6 years ago

vsuryamurthy commented 6 years ago

When I run the compute_bn_statistics.py, I get the following error: screenshot from 2018-04-01 19 46 43 I realize that the MaxTopBlobs() is set to 1 and it has to be increased to 3 to accommodate mean and variance blobs. @TimoSaemann Can you please help me with this problem?