Thank you for your reply @ptrblck
I set the linear layer in_features to 64 * 28 * 2 * 28 because the input image size is 120 * 120 * 120 and I have two conv layers. So how can I reduce the linear layer size and add another linear layer after the dopout layer?
Than you in advance.