normalize to (0.2,0.8) and different custom architecture