Add stage1 eval on dev during training