updated learning rate / switched architecture