So suppose I have 50 classes and I am using 3 layers is it sufficient?
And is the method for prediction given below sufficient…
net.eval()
    with torch.no_grad():
        outp=net(w)
        preds=torch.max(outp,1)
        x=preds[1][0]
        z="This is "+classes[x]
        say(z)
    return 0

assuming w is my single input image
Edit: Each class has 400 images of dimension 3x100x100