train heatnet