Small fix in albedo gen textfile + gradient predictor network added