trainer object added
change of plans perhaps
added strides to conv layer