Why the last layer in fully connected layer is 5?

[pmtang]

Thanks for the great project. I was wondering why you set the last layer to 5 in the fully connected network? Is there any reason to do that? I couldn't find explanation in the article also.
self.fc1 = nn.Sequential( nn.Linear(8*100*100, 500), nn.ReLU(inplace=True), nn.Linear(500, 500), nn.ReLU(inplace=True), nn.Linear(500, 5))

[Sujie1528]

I think it's just a design choice. Setting it to 10 or 15 would probably just work fine.