add dropout layer, fix slow training bug