add layerwised trainer