Yes correct, and for the test since I test each patch individually, the input size for linear layer  should be (1,864) and for CNN layer should be [1,1,11,11,7], like the thing that I used for training just now the batch size is 1