Some changes to training of nn. Check for lambda. More training
examples.

Change-Id: I2432397cfb98889f423d98e8d32a7974f211bf6e