Thank you for the reply.  Your code was v useful and helped me to identify the lr/learning rate parameter as the issue.  Interestingly, a several order of magnitude decrease in lr was needed to achieve good results when using a multi-layer vs single-layer model!  All good now, thanks again!