fixed model to have slightly larger l1 and to also take 4 channels of input, not 1