fix bug, try more learning epochs