When I train stage1 the model is not able to generate lower face heatmap. 