Stride can be adjusted for each conv layer
Parallelise the testing part (forget)