moved setting the number of layers/neurons to cond and pred modules