NaN loss when training regression network NaN loss when training regression network python python