Where to set #classes about enet-training HOT 7 CLOSED

You need to make changes only in loadCamvid.lua. What is the value of #classes after this line?

from enet-training.

It seems that no matter which dataset I use, the decoder part is as follow, which is only one layer of convolution with 13 filters. Do you have any idea where to fix?

th> model = torch.load('model-4.net')
th> model
nn.ConcatTable {
input
|-> (1): cudnn.SpatialConvolution(3 -> 13, 3x3, 2,2, 1,1) -> (2): nn.SpatialMaxPooling(2x2, 2,2)
... -> output
}

from enet-training.

Yes, the #classes is fine.

from enet-training.

As a sanity check consider looking at the models you're loading, and if they are pointing to the correct directories.

Since the dimensions are not hard-coded in the model/train/test files, this might be due to the pre-loaded model you're using.

from enet-training.

On closer inspection, I did find this in the encoder file pointing to a hard-coded dimension. This might fix your problem.

from enet-training.

See if you have the same issue as #3

from enet-training.

Thank you ishann and codeAC29, you were right, I was looking at wrong model. It works perfectly now!

from enet-training.